中文
相关论文

相关论文: Inference-time Alignment via Sparse Junction Steer…

200 篇论文

Small, highly trained, open-source large language models are widely used due to their inference efficiency, but further improving their quality remains a challenge. Sparse upcycling is a promising approach that transforms a pretrained dense…

机器学习 · 计算机科学 2024-11-15 Sasha Doubov , Nikhil Sardana , Vitaliy Chiley

Pretrained diffusion models are effective priors for Bayesian inverse problems, but posterior sampling with these priors is often costly because data-consistency guidance is applied throughout the full reverse trajectory. Existing methods…

机器学习 · 计算机科学 2026-05-29 Abduragim Shtanchaev , Albina Ilina , Yazid Janati , Arip Asadulaev , Martin Takac , Eric Moulines

This paper introduces a new sparse spatio-temporal structured Gaussian process regression framework for online and offline Bayesian inference. This is the first framework that gives a time-evolving representation of the interdependencies…

机器学习 · 统计学 2018-08-01 Danil Kuzin , Olga Isupova , Lyudmila Mihaylova

Attention-based models have revolutionized AI, but the quadratic cost of self-attention incurs severe computational and memory overhead. Sparse attention methods alleviate this by skipping low-relevance token pairs. However, current…

硬件体系结构 · 计算机科学 2026-01-13 Huizheng Wang , Hongbin Wang , Zichuan Wang , Zhiheng Yue , Yang Wang , Chao Li , Yang Hu , Shouyi Yin

Large language models spend most of their inference cost on attention over long contexts, yet empirical behavior suggests that only a small subset of tokens meaningfully contributes to each query. We formalize this phenomenon by modeling…

人工智能 · 计算机科学 2026-02-17 Vashista Nobaub

Inference-time steering enables pretrained diffusion/flow models to be adapted to new tasks without retraining. A widely used approach is the ratio-of-densities method, which defines a time-indexed target path by reweighting…

人工智能 · 计算机科学 2025-12-12 Ziseok Lee , Minyeong Hwang , Sanghyun Jo , Wooyeol Lee , Jihyung Ko , Young Bin Park , Jae-Mun Choi , Eunho Yang , Kyungsu Kim

Large Language Models (LLMs) trained for average correctness often exhibit mode collapse, producing narrow decision behaviors on tasks where multiple responses may be reasonable. This limitation is particularly problematic in ordinal…

人工智能 · 计算机科学 2026-02-04 Eric Yang , Jong Ha Lee , Jonathan Amar , Elissa Ye , Yugang Jia

Diffusion Transformers are fundamental for video and image generation, but their efficiency is bottlenecked by the quadratic complexity of attention. While block sparse attention accelerates computation by attending only critical key-value…

计算机视觉与模式识别 · 计算机科学 2026-02-04 Haopeng Li , Shitong Shao , Wenliang Zhong , Zikai Zhou , Lichen Bai , Hui Xiong , Zeke Xie

Inference-time scaling trades efficiency for increased reasoning accuracy by generating longer or more parallel sequences. However, in Transformer LLMs, generation cost is bottlenecked by the size of the key-value (KV) cache, rather than…

机器学习 · 计算机科学 2025-11-10 Adrian Łańcucki , Konrad Staniszewski , Piotr Nawrot , Edoardo M. Ponti

The growing environmental footprint of artificial intelligence (AI), especially in terms of storage and computation, calls for more frugal and interpretable models. Sparse models (e.g., linear, neural networks) offer a promising solution by…

机器学习 · 统计学 2025-09-23 Sylvain Sardy , Maxime van Cutsem , Xiaoyu Ma

Long-context models are essential for many applications but face inefficiencies in loading large KV caches during decoding. Prior methods enforce fixed token budgets for sparse attention, assuming a set number of tokens can approximate full…

机器学习 · 计算机科学 2025-02-19 Kan Zhu , Tian Tang , Qinyu Xu , Yile Gu , Zhichen Zeng , Rohan Kadekodi , Liangyu Zhao , Ang Li , Arvind Krishnamurthy , Baris Kasikci

DeepSeek Sparse Attention (DSA) sets the state of the art for fine-grained inference-time sparse attention by introducing a learned token-wise indexer that scores every prefix token and selects the most relevant ones for the main attention.…

机器学习 · 计算机科学 2026-05-11 Ruijie Zhou , Fanxu Meng , Yufei Xu , Tongxuan Liu , Guangming Lu , Muhan Zhang , Wenjie Pei

Steering methods influence Large Language Model behavior by identifying semantic directions in hidden representations, but are typically realized through inference-time activation interventions that apply a fixed, global modification to the…

计算与语言 · 计算机科学 2026-03-04 Chung-En Sun , Ge Yan , Zimo Wang , Tsui-Wei Weng

Recent advances in Large Language Models (LLMs) - particularly model scaling and test-time techniques - have greatly enhanced the reasoning capabilities of language models at the expense of higher inference costs. To lower inference costs,…

计算与语言 · 计算机科学 2025-11-21 Sangmook Lee , Dohyung Kim , Hyukhun Koh , Nakyeong Yang , Kyomin Jung

Attention is sparse in vision transformers. We observe the final prediction in vision transformers is only based on a subset of most informative tokens, which is sufficient for accurate image recognition. Based on this observation, we…

计算机视觉与模式识别 · 计算机科学 2021-10-27 Yongming Rao , Wenliang Zhao , Benlin Liu , Jiwen Lu , Jie Zhou , Cho-Jui Hsieh

Emerging from the pairwise attention in conventional Transformers, there is a growing interest in sparse attention mechanisms that align more closely with localized, contextual learning in the biological brain. Existing studies such as the…

机器学习 · 计算机科学 2025-03-12 Yuwei Sun , Hideya Ochiai , Zhirong Wu , Stephen Lin , Ryota Kanai

Gigantic pre-trained models have become central to natural language processing (NLP), serving as the starting point for fine-tuning towards a range of downstream tasks. However, two pain points persist for this paradigm: (a) as the…

机器学习 · 计算机科学 2023-05-25 Xuxi Chen , Tianlong Chen , Weizhu Chen , Ahmed Hassan Awadallah , Zhangyang Wang , Yu Cheng

Inference-time alignment for diffusion models aims to adapt a pre-trained reference diffusion model toward a target distribution without retraining the reference score network, thereby preserving the generative capacity of the reference…

机器学习 · 统计学 2026-02-05 Jinyuan Chang , Chenguang Duan , Yuling Jiao , Yi Xu , Jerry Zhijian Yang

Reward models (RMs) are critical components of alignment pipelines, yet they exhibit biases toward superficial stylistic cues, preferring better-presented responses over semantically superior ones. Existing debiasing methods typically…

计算与语言 · 计算机科学 2026-03-16 Mengyuan Sun , Zhuohao Yu , Weizheng Gu , Shikun Zhang , Wei Ye

Inference-time scaling has emerged as a powerful alternative to parameter scaling for improving language model performance on complex reasoning tasks. While existing methods have shown strong performance gains under fixed compute budgets,…

人工智能 · 计算机科学 2025-05-21 Giannis Chatziveroglou