Publications

Hanlin Zhang, Jikai Jin, Vasilis Syrgkanis, Sham Kakade (2026). Prescriptive Scaling Reveals the Evolution of Language Model Capabilities. arXiv.

PDF DOI ArXiv Blog Dataset

Jikai Jin, Kenneth Hung, Sanath Kumar Krishnamurthy, Baoyi Shi, Congshan Zhang (2026). Adaptive Exploration for Latent-State Bandits. arXiv.

Cite DOI ArXiv

Ayush Sawarni, Jikai Jin, Justin Whitehouse, Vasilis Syrgkanis (2026). Policy Learning with Abstention. In AISTATS 2026.

Cite ArXiv

Jikai Jin, Vasilis Syrgkanis (2026). Sharp Structure-Agnostic Lower Bounds for General Linear Functional Estimation. arXiv.

Cite DOI ArXiv

Jikai Jin, Lester Mackey, Vasilis Syrgkanis (2025). It's Hard to Be Normal: The Impact of Noise on Structure-agnostic Estimation . In NeurIPS 2025.

ArXiv

Jikai Jin, Vasilis Syrgkanis, Sham Kakade, Hanlin Zhang (2025). Discovering Hierarchical Latent Capabilities of Language Models via Causal Representation Learning. In arXiv preprint arXiv:2506.10378.

PDF Cite ArXiv

Jiayi Sheng, Luna Lyu, Jikai Jin, Tony Xia, Alex Gu, James Zou, Pan Lu (2025). Solving Inequality Proofs with Large Language Models. In NeurIPS 2025 (Datasets and Benchmarks track, spotlight).

PDF Cite ArXiv Website Twitter

Jikai Jin, Vasilis Syrgkanis (2025). Structure-agnostic Optimality of Doubly Robust Learning for Treatment Effect Estimation. In COLT 2025.

Cite ArXiv Poster Slides Blog

Jikai Jin, Vasilis Syrgkanis (2024). Learning Causal Representations from General Environments: Identifiability and Intrinsic Ambiguity. In NeurIPS 2024 (spotlight).

Cite ArXiv Slides

Kaifeng Lyu, Jikai Jin, Zhiyuan Li, Simon S. Du, Jason D. Lee, Wei Hu (2023). Dichotomy of Early and Late Phase Implicit Biases Can Provably Induce Grokking. In ICLR 2024.

Cite ArXiv

Jikai Jin, Zhiyuan Li, Kaifeng Lyu, Simon S. Du, Jason D. Lee (2023). Understanding Incremental Learning of Gradient Descent -- A Fine-grained Analysis of Matrix Sensing. In ICML 2023.

PDF Cite ArXiv Poster Slides

Jikai Jin, Yiping Lu, Jose Blanchet, Lexing Ying (2022). Minimax Optimal Kernel Operator Learning via Multilevel Training. In ICLR 2023 (spotlight).

PDF Cite ArXiv Slides Poster

Binghui Li, Jikai Jin, Han Zhong, John E. Hopcroft, Liwei Wang (2022). Why Robust Generalization in Deep Learning is Difficult: Perspective of Expressive Power. In NeurIPS 2022.

PDF Cite ArXiv

Jikai Jin, Suvrit Sra (2022). Understanding Riemannian Acceleration via a Proximal Extragradient Framework. In COLT 2022.

PDF Cite ArXiv

Jikai Jin, Bohang Zhang, Haiyang Wang, Liwei Wang (2021). Non-convex Distributionally Robust Optimization: Non-asymptotic Analysis. In NeurIPS 2021.

PDF Cite ArXiv

Bohang Zhang, Jikai Jin, Cong Fang, Liwei Wang (2020). Improved analysis of clipping algorithms for non-convex optimization. In NeurIPS 2020.

PDF Cite ArXiv