English
Related papers

Related papers: Group-SAE: Efficient Training of Sparse Autoencode…

200 papers

Sparse Autoencoders (SAEs) are a prominent tool in mechanistic interpretability (MI) for decomposing neural network activations into interpretable features. However, the aspiration to identify a canonical set of features is challenged by…

Machine Learning · Computer Science 2025-05-27 Xiangchen Song , Aashiq Muhamed , Yujia Zheng , Lingjing Kong , Zeyu Tang , Mona T. Diab , Virginia Smith , Kun Zhang

Sparse autoencoders are a promising new approach for decomposing language model activations for interpretation and control. They have been applied successfully to vision transformer image encoders and to small-scale diffusion models.…

Machine Learning · Computer Science 2025-07-14 Stepan Shabalin , Ayush Panda , Dmitrii Kharlapenko , Abdur Raheem Ali , Yixiong Hao , Arthur Conmy

Machine unlearning is a promising approach to improve LLM safety by removing unwanted knowledge from the model. However, prevailing gradient-based unlearning methods suffer from issues such as high computational costs, hyperparameter…

Machine Learning · Computer Science 2025-04-14 Aashiq Muhamed , Jacopo Bonato , Mona Diab , Virginia Smith

Deep learning has led to remarkable advances in computer vision. Even so, today's best models are brittle when presented with variations that differ even slightly from those seen during training. Minor shifts in the pose, color, or…

Computer Vision and Pattern Recognition · Computer Science 2022-10-25 Mark Ibrahim , Diane Bouchacourt , Ari Morcos

Fine-grained steering of language model outputs is essential for safety and reliability. Prompting and finetuning are widely used to achieve these goals, but interpretability researchers have proposed a variety of representation-based…

Computation and Language · Computer Science 2025-03-05 Zhengxuan Wu , Aryaman Arora , Atticus Geiger , Zheng Wang , Jing Huang , Dan Jurafsky , Christopher D. Manning , Christopher Potts

Surrogate-assisted evolutionary algorithms (SAEAs) are a key tool for addressing costly optimization tasks, with their efficiency being heavily dependent on the selection of surrogate models and infill sampling criteria. However, designing…

Neural and Evolutionary Computing · Computer Science 2025-07-08 Lindong Xie , Genghui Li , Zhenkun Wang , Edward Chung , Maoguo Gong

Sparse autoencoders are usually trained one layer at a time, even though transformer residual stream activations are strongly coupled across depth. This creates a practical problem for multi-layer interventions: different layerwise…

Machine Learning · Computer Science 2026-05-28 Prathyush Poduval , Calvin Yeung , Neel Desai , Mohsen Imani

Training large language models (LLMs) from scratch requires significant computational resources, driving interest in developing smaller, domain-specific LLMs that maintain both efficiency and strong task performance. Medium-sized models…

Computation and Language · Computer Science 2026-03-02 Chaitali Bhattacharyya , Hyunsei Lee , Junyoung Lee , Shinhyoung Jang , Il hong Suh , Yeseong Kim

Reward models (RMs) are a core component in the post-training of large language models (LLMs), serving as proxies for human preference evaluation and guiding model alignment. However, training reliable RMs under limited resources remains…

Artificial Intelligence · Computer Science 2025-11-12 Dengcan Liu , Jiahao Li , Zheren Fu , Yi Tu , Jiajun Li , Zhendong Mao , Yongdong Zhang

Sparse autoencoders (SAEs) are a technique for sparse decomposition of neural network activations into human-interpretable features. However, current SAEs suffer from feature absorption, where specialized features capture instances of…

Machine Learning · Computer Science 2025-09-29 Anton Korznikov , Andrey Galichin , Alexey Dontsov , Oleg Rogov , Elena Tutubalina , Ivan Oseledets

As large language models (LLMs) are increasingly deployed in real-world applications, the need to selectively remove unwanted knowledge while preserving model utility has become paramount. Recent work has explored sparse autoencoders (SAEs)…

Computation and Language · Computer Science 2026-04-28 Tomer Ashuach , Dana Arad , Aaron Mueller , Martin Tutek , Yonatan Belinkov

Predicting protein function from amino acid sequence remains a central challenge in data-scarce (low-$N$) regimes, limiting machine learning-guided protein design when only small amounts of assay-labeled sequence-function data are…

Machine Learning · Computer Science 2025-08-27 Darin Tsui , Kunal Talreja , Amirali Aghazadeh

Determining company similarity is a vital task in finance, underpinning risk management, hedging, and portfolio diversification. Practitioners often rely on sector and industry classifications such as SIC and GICS codes to gauge similarity,…

Audio pretrained models are widely employed to solve various tasks in speech processing, sound event detection, or music information retrieval. However, the representations learned by these models are unclear, and their analysis mainly…

Large language models (LLMs) encode a diverse range of linguistic features within their latent representations, which can be harnessed to steer their output toward specific target characteristics. In this paper, we modify the internal…

Computation and Language · Computer Science 2025-02-27 Sumanta Bhattacharyya , Pedram Rooshenas

Multimodal Large Language Models (MLLMs) have demonstrated remarkable capabilities by integrating visual and textual inputs, yet modality alignment remains one of the most challenging aspects. Current MLLMs typically rely on simple adapter…

Computer Vision and Pattern Recognition · Computer Science 2025-09-08 Yuanyang Yin , Yaqi Zhao , Yajie Zhang , Yuanxing Zhang , Ke Lin , Jiahao Wang , Xin Tao , Pengfei Wan , Wentao Zhang , Feng Zhao

Sparse autoencoders (SAEs) are widely used to extract sparse, interpretable latents from transformer activations. We test whether commonly used SAE quality metrics and automatic explanation pipelines can distinguish trained transformers…

Machine Learning · Computer Science 2026-01-28 Thomas Heap , Tim Lawson , Lucy Farnik , Laurence Aitchison

Pretraining large language models (LLMs) typically requires centralized clusters with thousands of high-memory GPUs (e.g., H100/A100). Recent decentralized training methods reduce communication overhead by employing federated optimization;…

Computation and Language · Computer Science 2026-05-05 Jinrui Zhang , Chaodong Xiao , Aoqi Wu , Xindong Zhang , Lei Zhang

Sparse Autoencoders (SAEs) are powerful tools for interpreting neural representations, yet their use in audio remains underexplored. We train SAEs across all encoder layers of Whisper and HuBERT, provide an extensive evaluation of their…

We propose a novel approach for training large language models (LLMs) to adhere to objectives defined within a latent embedding space. Our method leverages reinforcement learning (RL), treating a pre-trained LLM as an environment. Our…

Computation and Language · Computer Science 2024-10-29 Guy Tennenholtz , Yinlam Chow , Chih-Wei Hsu , Lior Shani , Ethan Liang , Craig Boutilier