Accelerating Sparse Autoencoder Training via Layer-Wise Transfer Learning in Large Language Models
Davide Ghilardi, Federico Belotti, Marco Molinari, Jaehyuk Lim
Abstract
Sparse AutoEncoders (SAEs) have gained popularity as a tool for enhancing the interpretability of Large Language Models (LLMs). However, training SAEs can be computationally intensive, especially as model complexity grows. In this study, the potential of transfer learning to accelerate SAEs training is explored by capitalizing on the shared representations found across adjacent layers of LLMs. Our experimental results demonstrate that fine-tuning SAEs using pre-trained models from nearby layers not only maintains but often improves the quality of learned representations, while significantly accelerating convergence. These findings indicate that the strategic reuse of pretrained SAEs is a promising approach, particularly in settings where computational resources are constrained.- Anthology ID:
- 2024.blackboxnlp-1.32
- Volume:
- Proceedings of the 7th BlackboxNLP Workshop: Analyzing and Interpreting Neural Networks for NLP
- Month:
- November
- Year:
- 2024
- Address:
- Miami, Florida, US
- Editors:
- Yonatan Belinkov, Najoung Kim, Jaap Jumelet, Hosein Mohebbi, Aaron Mueller, Hanjie Chen
- Venues:
- BlackboxNLP | WS
- SIG:
- Publisher:
- Association for Computational Linguistics
- Note:
- Pages:
- 530–550
- Language:
- URL:
- https://preview.aclanthology.org/jlcl-multiple-ingestion/2024.blackboxnlp-1.32/
- DOI:
- 10.18653/v1/2024.blackboxnlp-1.32
- Cite (ACL):
- Davide Ghilardi, Federico Belotti, Marco Molinari, and Jaehyuk Lim. 2024. Accelerating Sparse Autoencoder Training via Layer-Wise Transfer Learning in Large Language Models. In Proceedings of the 7th BlackboxNLP Workshop: Analyzing and Interpreting Neural Networks for NLP, pages 530–550, Miami, Florida, US. Association for Computational Linguistics.
- Cite (Informal):
- Accelerating Sparse Autoencoder Training via Layer-Wise Transfer Learning in Large Language Models (Ghilardi et al., BlackboxNLP 2024)
- PDF:
- https://preview.aclanthology.org/jlcl-multiple-ingestion/2024.blackboxnlp-1.32.pdf