arXiv Machine Learning By Inbasekaran S

EntroPrefill: Renyi-Guided Context Pruning with Conditional Stability Guarantees for Retrieval-Augmented Generation

Read the original on arXiv Machine Learning →

EntroPrefill introduces a Renyi‑guided context pruning method for retrieval‑augmented generation, coupling a proposal mechanism with constraints on discarded attention mass. The approach includes a sink‑isolated, regularized head pooling strategy that balances specialization and worst‑head coverage, and derives a computable upper bound on token removal along with finite‑sample observer guarantees for adaptive pruning layers. The paper also establishes a conditional transformer perturbation bound with Lipschitz constants, a first‑token decision‑margin corollary, and a counterexample illustrating limits of shallow observations, while outlining a systems analysis that distinguishes query‑head unions, physical page allocation, and KV‑transfer payload to provide an arithmetic break‑even condition for pruning.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Machine Learning.

arXiv Machine Learning
Sep 23

Magnitude Profile Pruning: Calibration-Free Structured Attention Head Removal for Transformer Compression

Magnitude Profile Pruning introduces a training‑free, calibration‑free method for removing attention heads in Transformer models by statistically detecting outliers in weight row norms. Heads whose projection weights fall within the bulk of the distribution are pruned, while outlier heads are retained. Across several models, the MP‑G variant achieves superior perplexity at various sparsity levels and yields significant parameter and FLOP reductions without requiring forward passes, calibration data, or gradient computations.

By Kasun Dewage, Marianna Pensky, Heranga K. Rathnasekara, Suranadi De Silva