Showing cs.LGShow all
2 papers · 1 filter
cs.LG2026
SinkRouter: Sink-Aware Routing for Efficient Long-Context Decoding in Large Language and Multimodal Models
Junnan Liu, Xinyan Liu, Peifeng Gao +4
In long-context decoding for LLMs and LMMs, attention becomes increasingly memory-bound because each decoding step must load a large amount of KV-cache data from GPU memory. Existi…
cs.LG2023
Towards Demystifying the Generalization Behaviors When Neural Collapse Emerges
Peifeng Gao, Qianqian Xu, Yibo Yang +5
Neural Collapse (NC) is a well-known phenomenon of deep neural networks in the terminal phase of training (TPT). It is characterized by the collapse of features and classifier into…