activity
20232026
most citedLayerSkip: Enabling Early Exit Inference and Self-Speculative Decoding

40 citations · 77 across the 8 of their papers we have counts for

collaborators
Showing 2026Show all

7 papers · 1 filter

cs.LG2026

Predicting Privileged Information Effectiveness for On-Policy Self-Distillation

Kaijian Zou, Emiliano Penaloza, MohammadHossein Rezaei +5

On-policy self-distillation (OPSD) enables large language models (LLMs) to learn from privileged information (PI) without a separate teacher model. However, prior work largely trea…

math.AC2026

Structural properties of admissible subgraphs and the depth of symbolic powers of cover ideals

Nguyen Thu Hang, Nguyen Thi Thanh Tam, Thanh Vu

Let be a simple graph with cover ideal in a polynomial ring $S = \k[x_1, \ldots, x_n]$. For each integer , let denote the maximum size of an ordered ma…

cs.AI2026

Root-Cause Attribution Is a Search Problem: Continual Search for Long-Horizon Agent Failures

Harsh Raj, David Lee, Anas Mahmoud +7

The increasing deployment of AI agents in long-horizon tasks yields massive execution logs. Diagnosing failures within these records is crucial for reliability, as it transforms ou…

cs.AI2026

Model or Harness? An Interaction-Centric Taxonomy for Localizing Agent Failures

Harsh Raj, Vipul Gupta, Anas Mahmoud +4

Existing evaluations often reduce agent failures to system-level outcomes, obscuring where the fault originated and which intervention would improve the agent system. This creates…

cs.LG2026

Rubric-Guided Self-Distillation: Post-Training Without Rubric Verifiers

MohammadHossein Rezaei, Anas Mahmoud, Zihao Wang +6

Rubrics have emerged as an alternative to RLVR in open-ended domains where a single ground-truth final answer is not available. Existing rubric-based training methods rely on an LL…

cs.AI2026

Not Every Rubric Teaches Equally: Policy-Aware Rubric Rewards for RLVR

Utkarsh Tyagi, Xingang Guo, MohammadHossein Rezaei +5

Reinforcement learning with verifiable rewards has made post-training highly effective when correctness can be checked automatically. However, many important model behaviors requir…

cs.AI2026

Reward Hacking in Rubric-Based Reinforcement Learning

Anas Mahmoud, MohammadHossein Rezaei, Zihao Wang +3

Reinforcement learning with verifiable rewards has enabled strong post-training gains in domains such as math and coding, though many open-ended settings rely on rubric-based rewar…