CLIFF: Continual Learning for Incremental Flake Features in 2D Material Identification
2508.17261v1
cs.CV, cs.LG
2025-08-27
Авторы:
Sankalp Pandey, Xuan Bac Nguyen, Nicholas Borys, Hugh Churchill, Khoa Luu
Резюме на русском
#### Контекст
Идентификация квантовых флейсов является ключевым вопросом для развития скалярных квантовых технологий, но автоматическая классификация слоёв на основе оптического микроскопии сталкивается с значительными трудностями из-за выраженных визуальных изменений между различными материалами. Для решения этой проблемы требуются мощные методы, способные обучаться постепенно и адаптироваться к новым материалам без полной переобучения. Наша проблема лежит в центре области континюального обучения, где требуется создать модели, которые могут эффективно обобщаться на новых данных без удаления информации о предыдущих. Особенно актуальным это становится в контексте двумерных (2D) материалов, где природа физических и оптических свойств требует точности и гибкости.
#### Метод
Мы предлагаем Continual-Learning Framework for Flake Layer Classification (CLIFF), ориентированный на точность и эффективность в классификации слоёв квантовых материалов. Фреймворк основывается на трёх основных компонентах:
1. **Material-Specific Prompts и Embeddings**: CLIFF использует пространственные модели, описывающие физические свойства материалов. Эти модели адаптируются для каждого нового материала с помощью материал-специфических предложений (prompts) и вспомогательных входных данных (embeddings).
2. **Delta Head и Prompt Pool**: CLIFF обучает отдельный "delta head" для каждого материала, используя его для выявления отличий по отношению к предыдущим материалам. Для этого используется пул предложений (prompt pool), который помогает быстро ориентироваться в новых условиях.
3. **Memory Replay и Knowledge Distillation**: Для борьбы с проблемой кратковременного забывания (forgetting), CLIFF включает механизм памяти реплея, позволяющий модели повторно просматривать ранее натренированные данные, а также использовать техники knowledge distillation для улучшения устойчивости модели.
CLIFF является первым систематическим подходом к континюальному обучению в 2D-материалах, объединяя принципы continual learning и memory-based techniques.
#### Результаты
Мы провели эксперименты на наборе данных, содержащих слои различных квантовых материалов, как известных, так и новых. Наша модель CLIFF показала высокую точность классификации и существенно меньшую степень забывания по сравнению с традиционным одиночным обучением (fine-tuning) и базовым методом с подсказками (prompt-based baseline). Также мы проанализировали влияние размера памяти реплирования и количества материалов на полученные результаты. Эксперименты показали, что CLIFF обеспечивает значительно более стабильное обучение и эффективность при работе с новыми материалами.
#### Значимость
CLIFF открывает новые возможности для автоматизированной классификации квантовы
Abstract
Identifying quantum flakes is crucial for scalable quantum hardware; however,
automated layer classification from optical microscopy remains challenging due
to substantial appearance shifts across different materials. In this paper, we
propose a new Continual-Learning Framework for Flake Layer Classification
(CLIFF). To our knowledge, this is the first systematic study of continual
learning in the domain of two-dimensional (2D) materials. Our method enables
the model to differentiate between materials and their physical and optical
properties by freezing a backbone and base head trained on a reference
material. For each new material, it learns a material-specific prompt,
embedding, and a delta head. A prompt pool and a cosine-similarity gate
modulate features and compute material-specific corrections. Additionally, we
incorporate memory replay with knowledge distillation. CLIFF achieves
competitive accuracy with significantly lower forgetting than naive fine-tuning
and a prompt-based baseline.
Ссылки и действия
Дополнительные ресурсы: