Sparse-Autoencoder-Guided Internal Representation Unlearning for Large Language Models

## Контекст Современные большие языковые модели (LLMs) находят широкое применение в различных областях, но при этом сталкиваются с значительными проблемами с точки зрения конфиденциальности и авторских прав. Одна из основных проблем заключается в том, что LLMs могут сохранять внутренние представления, которые могут быть использованы для восстановления конфиденциальной информации. Это вызывает необходимость в разработке эффективных методов "учтения" (unlearning), которые позволят удалить упоминания конкретных целевых сущностей из модели без существенного ущерба для остальных знаний. Несмотря на наличие многих методов знаний о сущностях, большинство из них стремятся уменьшить вероятность появления ненужных ответов, а не удалить конкретные знания. Это приводит к проблеме "спонтанного возникновения" (model collapse) и неверной подавленности знаний. ## Метод Мы предлагаем новую методологию "учтения", основанную на использовании запароленного автокодировщика (sparse autoencoder). Целью этой методики является изменение внутренних активаций модели таким образом, чтобы активации целевой сущности стали неотличимыми от активаций неизвестных сущностей. Мы определяем "учтение" как состояние, в котором модель не может отличить данную сущность от неизвестного. Данный подход предлагает прямое воздействие на внутренние представления модели, а не только подавление вывода. Мы строим целевую функцию, целью которой является минимизация различия внутренних активаций целевой сущности от неизвестных, а также максимизация различий от других известных сущностей. Это позволяет достичь желаемого эффекта "учтения" без потери других знаний. ## Результаты Мы проверили нашу методику на нескольких тестовых наборах данных и показали, что она эффективно удаляет знания о конкретных сущностях из модели без привлечения дополнительных данных или дополнительной обучающей стадии. Мы измерили эффективность наших результатов на показателе удаления знаний и показали, что наш подход позволяет избежать модели collapse, который часто возникает при других методах. Кроме того, мы проанализировали навыки модели в задачах вопроса-ответа и показали, что наш подход эффективно снижает вероятность появления нежелательных ответов без существенной потери невинных знаний. ## Значимость Наш подход может быть применен в различных областях, где требуется удаление конфиденциальных данных из моделей, например, в сфере здравоохранения, финансов или правосудия. Он обеспечивает более надежное удаление знаний по сравнению с другими методами, а также уменьшает вероятность модели collapse. Благодаря тому, что метод оптимизирует внутренние