5 papers
Expanding LLM Reasoning
Rian Atri, Evan Luo
Extra inference compute is usually spent on sampling more reasoning chains. We study where inside an existing chain an additional continuation should begin. We define expansion uti…
What Does an Observability Foundation Model Know?
Dhyey Dharmendrakumar Mavani, Rian Atri, Tairan Ji
A linear probe can show that a label is recoverable from a model's hidden states, but not whether that goes beyond what the input already reveals, or whether the model uses it. We…
Efficient Reasoning at Fixed Test-Time Cost via Length-Aware Attention Priors and Gain-Aware Training
Rian Atri
We study efficient reasoning under tight compute. We ask how to make structured, correct decisions without increasing test time cost. We add two training only components to small a…
Deterministic Fuzzy Triage for Legal Compliance Classification and Evidence Retrieval
Rian Atri
Legal teams increasingly use machine learning to triage large volumes of contractual evidence, but many models are opaque, non-deterministic, and difficult to align with frameworks…
PyHealth 2.0: A Comprehensive Open-Source Toolkit for Accessible and Reproducible Clinical Deep Learning
John Wu, Yongda Fan, Zhenbang Wu +14
Difficulty replicating baselines, high computational costs, and required domain expertise create persistent barriers to clinical AI research. To address these challenges, we introd…