α-β order denotes alphabetical authorship ordering, and * denotes equal contribution.

Theory & Mechanism of Deep Learning

• Neural Networks Provably Learn Spectral Representations for Group Composition.
  Jianliang He*, Leda Wang*, Fengzhuo Zhang, Siyu Chen, and Zhuoran Yang.
  Submitted, arXiv.2606.02993, 2026. [arXiv][pdf] [code][slides]
• On the Mechanism and Dynamics of Modular Addition: Fourier Features, Lottery Ticket, and Grokking.
  Jianliang He, Leda Wang, Siyu Chen, and Zhuoran Yang.
  Submitted, arXiv.2602.16849, 2026. [arXiv][pdf] [code] [blog]
• How Transformers Learn Causal Structures In-Context: Explainable Mechanism Meets Theoretical Guarantee.
  Jianzhe Wei, Siyu Chen, Jianliang He, and Zhuoran Yang.
  International Conference on Learning Representations (ICLR), 2026. [pdf]
• In-Context Linear Regression Demystified: Training Dynamics and Mechanistic Interpretability of Multi-Head Softmax Attention.
  Jianliang He, Xintian Pan, Siyu Chen, and Zhuoran Yang.
  International Conference on Machine Learning (ICML), 2025. [arXiv][pdf] [code] [blog]

Statistical Machine Learning

• From Words to Actions: Unveiling the Theoretical Underpinnings of LLM-Driven Autonomous Systems.
  Jianliang He*, Siyu Chen*, Fengzhuo Zhang, and Zhuoran Yang.
  International Conference on Machine Learning (ICML), 2024. [arXiv][pdf]
• Sample-efficient Learning of Infinite-horizon Average-reward MDPs with General Function Approximation.
  Jianliang He, Han Zhong, and Zhuoran Yang.
  International Conference on Learning Representations (ICLR), 2024. [arXiv][pdf]
• Data-light Uncertainty Set Merging with Admissibility.
  Shenghao Qin*, Jianliang He*, Qi Kuang*, Bowen Gang, and Yin Xia.
  Submitted, arXiv:2410.12201, 2024. [arXiv][pdf]
• Harnessing the Collective Wisdom: Fusion Learning using Decision Sequences from Diverse Sources.
  (α-β order) Trambak Banerjeea, Bowen Gang, and Jianliang He.
  Biometrika, 2026. [arXiv][pdf]

Large Language Model

• Relative Generalization Invariance of LLM Pretraining.
  
  Fengzhuo Zhang, Shuche Wang, Shenggui Li, Tianyu Ruan, Jianliang He, Ivor Tsang, Tianyu Pang, Chao Du, Tianwei Zhang, and Zhuoran Yang.
  Submitted, arXiv:2609.33016, 2026. [arXiv] [pdf]
• Antares: Foundation Models for Agentic Vulnerability Localization. [HuggingFace]
  
  Foundation AI, Cisco.

  Full authors

  Supriti Vijay, Aman Priyanshu, Didier Chapoteau, Arthur Goldblatt, Jianliang He, Kimia Majd, Fraser Burch, Baturay Saglam, Takahiro Matsumoto, Zhuoran Yang, and Amin Karbasi.

  Technical Report, 2026. [arXiv] [pdf]
  • Vulnerability Localization Benchmark, 2026. [arXiv] [pdf]
    Full authors

    Aman Priyanshu, Supriti Vijay, Kimia Majd, Xuhong He, Fraser Burch, Takahiro Matsumoto, Jianliang He, Baturay Saglam, Arthur Goldblatt, Zhuoran Yang, and Amin Karbasi.

• Llama 3.1 Foundation AI Security LLM 8B Technical Report Series. [HuggingFace]
  
  Foundation AI, Cisco.
  • Reasoning, Technical Report, 2026. [arXiv] [pdf]
    Full authors

    Zhuoran Yang, Ed Li, Jianliang He, Aman Priyanshu, Baturay Saglam, Paul Kassianik, Sajana Weerawardhena, Anu Vellore, Blaine Nelson, Neusha Javidnia, Arthur Goldblatt, Fraser Burch, Avi Zohary, Assaf Eisenman, Mahdi Sabbaghi, Supriti Vijay, Rahim Dharssi, Dhruv Kedia, Kojin Oshiba, Yaron Singer, and Amin Karbasi.

  • Instruct, Technical Report, 2025. [arXiv] [pdf]
    Full authors

    Sajana Weerawardhena, Paul Kassianik, Blaine Nelson, Baturay Saglam, Anu Vellore, Aman Priyanshu, Supriti Vijay, Massimo Aufiero, Arthur Goldblatt, Fraser Burch, Ed Li, Jianliang He, Dhruv Kedia, Kojin Oshiba, Zhouran Yang, Yaron Singer, and Amin Karbasi.

Miscellaneous

• False Discovery Control in Multiple Testing: A Selective Overview of Theories and Methodologies. [pdf]
  Short Version as a Book Chapter in ICSA Book Series in Statistics: Big Data Analytics in Biostatistics and Bioinformatics.