About Me

I earned my Ph.D. (2021-2025) from the University of Liverpool, under the supervision of Professor Xiaowei Huang and Professor Xinping Yi. My current research focus on the following topics:

  • Efficient AI (e.g., Hardware-Aligned Linear/Quasilinear Scalable Transformers)
  • (Non-Euclidean) Geometric Deep Learning (e.g., Hyperbolic Graph Embedding)
  • Human-AI Preference Alignment (e.g., RLHF, Red-teaming and Defence).

Before Liverpool, I received my dual bachelor’s degree (2017-2021) in Computer Science from both Australian National University and Shandong University.

From October 2025, I will be joining the Imperial College London as a postdoctoral researcher.

Recent News

  • (06/2025) One paper accepted to Transactions on Machine Learning Research.
  • (01/2025) One paper accepted to ICLR 2025.
  • (08/2024) One paper accepted to IJCAI 2024.
  • (08/2024) One paper accepted to ECML-PKDD 2024.
  • (06/2024) One paper accepted to IEEE Transactions on Artificial Intelligence.
  • (04/2024) One paper accepted to ICASSP 2024.
  • (03/2024) One paper accepted to AAAI 2024.

Selected Publications

TMLR
sym

Toward Linearly Regularizing the Geometric Bottleneck of Linear Generalized Attention

Jiaxu Liu, Xinping Yi, Xiangyu Yin, Yuhang Song, Gaojie Jin, Xiaowei Huang

ECML
sym

Continuous Geometry-Aware Graph Diffusion via Hyperbolic Neural PDE

Jiaxu Liu, Xinping Yi, Sihao Wu, Xiangyu Yin, Tianle Zhang, Xiaowei Huang, Shi Jin

IEEE Trans. AI
sym

Full list of my articles on my Google Scholar profile.

Academic Service

  • Program Committee for AAAI 2025
  • Invited Reviewer for NeurIPS 2024, 2025
  • Invited Reviewer for ICLR 2025
  • Invited Reviewer for Frontiers in Neurorobotics

Teaching

I was involved as a TA in the following courses:

  • University of Liverpool: COMP338 (Computer Vision), ELEC230 (Robotics), COMP534 (Applied Artificial Intelligence)
  • Australian National University: COMP3600/6466 (Algorithms)