Ax Zhitong Gao, Parham Rezaei, Ali Cy, Mingqiao Ye, Nata\v{s}a Jovanovi\'c, Jesse Allardice, Afshin Dehghan, Amir Zamir, Roman Bachmann, O\u{g}uzhan Fatih Kar 4/20/2026

(1D) Ordered Tokens Enable Efficient Test-Time Search

Study on ordered tokenization enabling efficient test-time search in autoregressive generative models through token structure optimization.

Ax Guy Kaplan, Zorik Gekhman, Zhen Zhu, Lotem Rozner, Yuval Reif, Swabha Swayamdipta, Derek Hoiem, Roy Schwartz 4/20/2026

Why Fine-Tuning Encourages Hallucinations and How to Fix It

Study on how supervised fine-tuning increases LLM hallucinations through exposure to new facts and mitigation using continual learning techniques.

Ax Constantinos Karouzos, Xingwei Tan, Nikolaos Aletras 4/20/2026

Where does output diversity collapse in post-training?

Analysis of output diversity collapse in post-trained language models, showing models produce less varied outputs than base versions, affecting inference-time scaling.

Ax Yunhe Li, Hao Shi, Bowen Deng, Wei Wang, Mengzhe Ruan, Hanxu Hou, Zhongxiang Dai, Siyang Gao, Chao Wang, Shuang Qiu, Linqi Song 4/20/2026

Learning to Reason with Insight for Informal Theorem Proving

arXiv paper proposing framework for informal theorem proving with LLMs using insight-driven reasoning.

Ax Luke Braithwaite, Alessio Borgi, Gabriele Onorato, Kristjan Tarantelli, Iulia Duta, Francesco Restuccia, Fabrizio Silvestri, Pietro Li\`o 4/20/2026

Heterogeneous Sheaf Neural Networks

arXiv paper proposing HetSheaf, a framework for heterogeneous graph neural networks across different node/edge types.

Ax Daniel Jenson, Jhonathan Navott, Mengyan Zhang, Makkunda Sharma, Elizaveta Semenova, Seth Flaxman 4/20/2026

Transformer Neural Processes - Kernel Regression

arXiv paper on HetSheaf framework for heterogeneous graph neural networks supporting multiple node/edge types in real-world applications.

Ax Witold Wydma\'nski, Marek \'Smieja 4/20/2026

AutoNFS: Automatic Neural Feature Selection

arXiv paper on Few-Shot Preference Optimization (FSPO) for personalizing LLMs using meta-learning on synthetic preference data.

Ax Zhucong Li, Powei Chang, Jin Xiao, Zhijian Zhou, Qianyu He, Jiaqing Liang, Fenglei Cao, Xu Yinghui, Yuan Qi 4/20/2026

ChemAmp: Amplified Chemistry Tools via Composable Agents

ChemAmp: Framework for composable LLM agents in chemistry using tool amplification to enhance multi-tool orchestration capabilities.

Ax Paul Janson, Benjamin Therien, Quentin Anthony, Xiaolong Huang, Abhinav Moudgil, Eugene Belilovsky 4/20/2026

PyLO: Towards Accessible Learned Optimizers in PyTorch

PyLO: PyTorch package making learned optimizers accessible, providing drop-in replacements for standard optimizers like Adam.