arXiv Machine Learning By Lin Zhang

SAKI-RoPE: Score-Aware Low-Rank Key Indexing with RoPE-Aware Covariance and Random-Matrix Noise Analysis

Read the original on arXiv Machine Learning →

The paper introduces SAKI-RoPE, a training‑free low‑rank key‑indexing method that optimally preserves attention scores by solving a covariance‑weighted rank‑r problem using query and key statistics. It outperforms key‑PCA across multiple large language models and context lengths, achieving higher recall with significantly lower rank. The authors also analyze random‑matrix noise to justify the calibration token budget and demonstrate that clipping the Marchenko‑Pastur bulk improves performance when calibration data is limited.

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 Computation and Language
Oct 1

Listening to the Wise Few: Query-Key Alignment Unlocks Latent Correct Answers in Large Language Models

arXiv:2410.02343v2 Announce Type: replace Abstract: Large language models (LLMs) routinely fail to output the correct option in multiple-choice question answering (MCQA) while encoding the answer int...

By Eduard Tulchinskii, Kristian Kuznetsov, Laida Kushnareva, Anastasia Voznyuk, Andrei Andriiainen, Irina Piontkovskaya, Evgeny Burnaev, Serguei Barannikov