He, Jianliang

4 publications

ICLR 2026 How Transformers Learn Causal Structures In-Context: Explainable Mechanism Meets Theoretical Guarantee Jianzhe Wei, Siyu Chen, Jianliang He, Zhuoran Yang

ICML 2025 In-Context Linear Regression Demystified: Training Dynamics and Mechanistic Interpretability of Multi-Head SoftMax Attention Jianliang He, Xintian Pan, Siyu Chen, Zhuoran Yang

ICML 2024 From Words to Actions: Unveiling the Theoretical Underpinnings of LLM-Driven Autonomous Systems Jianliang He, Siyu Chen, Fengzhuo Zhang, Zhuoran Yang

ICLR 2024 Sample-Efficient Learning of Infinite-Horizon Average-Reward MDPs with General Function Approximation Jianliang He, Han Zhong, Zhuoran Yang