DAGFormer: Predicted Cross-Layer Topologies Cut Perplexity Across Model Scales

AI-Interpretability-illinois · hf · 2026-10-19

An Illinois paper introduces DAGFormer, improving Transformer efficiency: instead of every layer reading the previous one, a jointly trained 2-layer causal Transformer topology predictor plans Q/K/V/residual cross-layer connections per token, forming a global DAG over read sources. At 75M, 300M and 1B scales, the planned-topology variants with refinement reduce language modeling perplexity and boost downstream performance, with an interactive visualization of the trained graphs.

Original post →

More from Research

Research channel →