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.

Ссылки и действия