Ax Oscar Hill, Mateo Espinosa Zarlenga, Mateja Jamnik 3/2/2026

Hierarchical Concept-based Interpretable Models

Hierarchical Concept Embedding Models improve interpretability of deep neural networks by mapping inputs to human-interpretable concept representations with inter-concept relationships.

Ax Xianglong Shi, Ziheng Chen, Yunhan Jiang, Nicu Sebe 3/2/2026

Intrinsic Lorentz Neural Network

Proposes Intrinsic Lorentz Neural Network for fully intrinsic hyperbolic geometry operations on hierarchical data representations.

Ax Roy Betser, Eyal Gofer, Meir Yossef Levi, Guy Gilboa 3/2/2026

InfoNCE Induces Gaussian Distribution

Theoretical analysis showing InfoNCE contrastive loss induces Gaussian structure in learned representations for foundation models.

Ax Daniel Yang, Samuel Stante, Florian Redhardt, Lena Libon, Parnian Kassraie, Ido Hakimi, Barna P\'asztor, Andreas Krause 3/2/2026

RewardUQ: A Unified Framework for Uncertainty-Aware Reward Models

Proposes RewardUQ framework for uncertainty-aware reward models in LLM alignment that reduces annotation costs and prevents overoptimization.

Ax Ali Behrouz, Zeman Li, Yuan Deng, Peilin Zhong, Meisam Razaviyayn, Vahab Mirrokni 3/2/2026

Memory Caching: RNNs with Growing Memory

Memory caching architecture enabling RNNs with growing memory capacity and subquadratic complexity as alternative to Transformers for sequence modeling.

Ax Weinan Dai, Hanlin Wu, Qiying Yu, Huan-ang Gao, Jiahao Li, Chengquan Jiang, Weiqiang Lou, Yufan Song, Hongli Yu, Jiaze Chen, Wei-Ying Ma, Ya-Qin Zhang, Jingjing Liu, Mingxuan Wang, Xin Liu, Hao Zhou 3/2/2026

CUDA Agent: Large-Scale Agentic RL for High-Performance CUDA Kernel Generation

Agentic RL system using LLMs for high-performance CUDA kernel generation at scale, outcompeting traditional compiler-based approaches.