activity
20242026
most citedEfficient Causal Graph Discovery Using Large Language Models

6 citations · 6 across the 2 of their papers we have counts for

collaborators

7 papers

cs.LG2026

Auto-Discovery-Bench: Diagnosing Structured State Tracking in Oracle-Guided Discovery

Tingting Chen, Beibei Lin, Srinivas Anumasa +5

Interactive discovery requires agents to maintain and update structured beliefs over many rounds of feedback. Before evaluating agents in noisy, open-ended scientific environments,…

cs.LG20266 cited

Efficient Causal Graph Discovery Using Large Language Models

Thomas Jiralerspong, Xiaoyin Chen, Yash More +2

We propose a novel framework that leverages LLMs for full causal graph discovery. While previous LLM-based methods have used a pairwise query approach, this requires a quadratic nu…

cs.LG2026

A Comedy of Estimators: On KL Regularization in RL Training of LLMs

Vedant Shah, Johan Obando-Ceron, Vineet Jain +10

The reasoning performance of large language models (LLMs) can be substantially improved by training them with reinforcement learning (RL). The RL objective for LLM training involve…

cs.LG2026

Recursive Self-Aggregation Unlocks Deep Thinking in Large Language Models

Siddarth Venkatraman, Vineet Jain, Sarthak Mittal +9

Test-time scaling methods improve the capabilities of large language models (LLMs) by increasing the amount of compute used during inference to make a prediction. Inference-time co…

cs.LG2025

Masked Generative Priors Improve World Models Sequence Modelling Capabilities

Cristian Meo, Mircea Lica, Zarif Ikram +6

Deep Reinforcement Learning (RL) has become the leading approach for creating artificial agents in complex environments. Model-based approaches, which are RL methods with world mod…

cs.AI2025

AI-Assisted Generation of Difficult Math Questions

Vedant Shah, Dingli Yu, Kaifeng Lyu +8

Current LLM training positions mathematical reasoning as a core capability. With publicly available sources fully tapped, there is unmet demand for diverse and challenging math que…