Ax Shuofei Qiao, Yunxiang Wei, Jiazheng Fan, Bin Wu, Busheng Zhang, Mengru Wang, Yuqi Zhu, Ningyu Zhang, Keyan Ding, Qiang Zhang, Huajun Chen 5/25/2026

SciAtlas: A Large-Scale Knowledge Graph for Automated Scientific Research

SciAtlas: large-scale knowledge graph for scientific research enabling topological reasoning and interdisciplinary integration for AI agents.

Ax Shubham Parashar, Atharv Chagi, Jacob Helwig, Lakshmi Jotsna, Sushil Vemuri, James Caverlee, Dileep Kalathil, Shuiwang Ji 5/25/2026

Learnability-Informed Fine-Tuning of Diffusion Language Models

Novel fine-tuning approach for diffusion language models that accounts for token learnability to improve reasoning capabilities.

Ax Colin Gaffney, Shutong Li, Daniel Ng, Anastasia Petrushkina, Niket Kumar, Adam Cogdell, Mridul Sahu, Yaning Liang, Nikhil Bansal, Justin Pan, Angel Mau, Abhishek Agrawal, Marco Berlot, Ruoxin Sang, Kiranbir Sodhia, Rakesh Iyer 5/25/2026

Orbax: Distributed Checkpointing with JAX

Orbax: JAX-native distributed checkpointing library abstracting complexity of accelerator systems for ML model lifecycle management.

Ax Philipp Kern, Lorenzo Rovida, Samuel Teuber, Edoardo Manino, Carsten Sinz, Alberto Leporati 5/25/2026

Encrypted Neural Networks without Overflows

Methods for evaluating neural networks on encrypted data using fully homomorphic encryption without overflow issues in CKKS scheme.

Ax Caleb Skinner, Yihan Guo, Meng Li 5/25/2026

LLM Sparsity Prior for Robust Feature Selection

Framework for using LLMs to generate domain-informed priors for feature selection, with methods to quantify weight quality and improve robustness.

Ax Baturay Saglam, Dionysis Kalogerias 5/25/2026

Self-Improving In-Context Learning

Method for optimizing few-shot prompt embeddings at test time using log-probabilities signal to improve in-context learning performance.

Ax Mario Markov (INSAIT, Sofia University "St. Kliment Ohridski"), Stefan Maria Ailuro (INSAIT, Sofia University "St. Kliment Ohridski"), Mohammad Mahdi (INSAIT, Sofia University "St. Kliment Ohridski"), Luc Van Gool (INSAIT, Sofia University "St. Kliment Ohridski"), Danda Pani Paudel (INSAIT, Sofia University "St. Kliment Ohridski") 5/25/2026

B-GRTO: Bootstrapped Group Relative Tool Optimization for Referring Segmentation

Bootstrapped group relative tool optimization for referring segmentation using vision-language models with reinforcement learning.

Ax Luke Braithwaite, Alessio Borgi, Gabriele Onorato, Kristjan Tarantelli, Francesco Restuccia, Fabrizio Silvestri, Pietro Li\`o 5/25/2026

Heterogeneous Sheaf Neural Networks

Heterogeneous graph neural networks for handling nodes/edges with different types and feature spaces across domains.

Ax Zhong Li, Qi Huang, Lincen Yang, Jiayang Shi, Zhao Yang, Niki van Stein, Thomas B\"ack, Matthijs van Leeuwen 5/25/2026

Diffusion and Flow Matching Models for Tabular Data: A Survey

Survey of diffusion and flow matching models for tabular data generation, addressing missing values and domain constraints.