Ax Haowen Wang, Yaxin Du, Jian Yang, Jiajun Wu, Shukai Liu, Yuxuan Zhang, Pingjie Wang, Siheng Chen, Tuney Zheng, Ming Zhou, Xianglong Liu, Bryan Dai 6/1/2026

MIRA: Mid-training Rubric Anchoring for Source-Aware Data Selection

MIRA method for source-aware mid-training data selection balancing pretraining and downstream capability optimization in LLMs.

Ax Shijie Wang, Jiani Huang, Zhikai Chen, Yu Song, Wenzhuo Tang, Haitao Mao, Wenqi Fan, Hui Liu, Xiaorui Liu, Dawei Yin, Qing Li 6/1/2026

Graph Machine Learning in the Era of Large Language Models (LLMs)

Survey of graph machine learning integration with large language models for tasks in social networks and knowledge graphs.

Ax Diane Tchuindjo, Omar Khattab 6/1/2026

Reasoning-Intensive Regression

Introduces reasoning-intensive regression using LLMs to deduce subtle numerical scores from text for rubric-based scoring and reward modeling.

Ax Chenhui Hu, Pengfei Cao, Yubo Chen, Kang Liu, Jun Zhao 6/1/2026

Towards Atoms of Large Language Models

Proposes Atom Theory to define and identify fundamental representational units in LLMs using non-Euclidean atomic inner product metric.

Ax Aditya Thimmaiah, Jiyang Zhang, Jayanth Srinivasa, Junyi Jessy Li, Milos Gligoric 6/1/2026

LLMs Lean on Priors, Not Programming Language Semantics

Investigates whether LLMs condition reasoning on formal programming semantics or learned statistical priors through systematic program execution analysis.

Ax Abdelkrim Zitouni, Mehdi Hennequin, Juba Agoun, Ryan Horache, Nadia Kabachi, Omar Rivasplata 6/1/2026

PAC-Bayesian Reinforcement Learning Trains Generalizable Policies

Derives PAC-Bayesian generalization bounds for reinforcement learning accounting for Markov dependencies, enabling non-vacuous certificates for modern RL.

Ax Dong Jing, Gang Wang, Jiaqi Liu, Weiliang Tang, Zelong Sun, Yunchao Yao, Zhenyu Wei, Yunhui Liu, Zhiwu Lu, Mingyu Ding 6/1/2026

Mixture of Horizons in Action Chunking

Studies action chunk length trade-offs in vision-language-action robotic models, proposes mixture of horizons approach for improved performance.