Jinrui Liu

AIR: Post-training Data Selection for Reasoning via Attention Head Influence

LLMs achieve remarkable multi-step reasoning capabilities, yet effectively transferring these skills via post-training distillation remains challenging. Existing data selection methods, ranging from manual curation to heuristics based on length, entropy, or overall loss, fail to capture the causal i…