Raw Data Matters: Enhancing Prompt Tuning by Internal Augmentation on Vision-Language Models

2508.02671v1 cs.CV 2025-08-05
Авторы:

Haoyang Li, Liang Wang, Chao Wang, Siyu Zhou, Jing Jiang, Yan Peng, Guodong Long

Резюме на русском

КОНТЕКСТ И ПРОБЛЕМАТИКА Современные модели обработки изображений и текста, такие как CLIP (Contrastive Language–Image Pretraining), становятся все более важными в области искусственного интеллекта благодаря их способности эффективно работать с мультидоменными данными. Эти модели обучаются на огромных наборах данных, содержащих пары изображений и текстов, что позволяет им извлекать и связывать семантическую информацию из различных источников. Несмотря на их успешность, существует ряд проблем, связанных с адаптацией и тонкой настройкой таких моделей для конкретных задач или датасетов. Одной из ключевых проблем является высокая стоимость и сложность сбора и обработки дополнительных данных для повышения производительности моделей. Это особенно актуально в контексте настройки подсказок (prompt tuning), где задача состоит в том, чтобы оптимально адаптировать модель к конкретной задаче, используя ограниченные ресурсы. Существующие стратегии увеличения данных для настройки подсказок часто опираются на внешние источники знаний, такие как большие языковые модели или предварительно структурированные базы знаний. Эти подходы, хотя и эффективные, требуют значительных затрат на сбор и обработку данных, а также могут игнорировать потенциал использования внутренних признаков изображений, уже имеющихся в исходном наборе данных. Таким образом, возникает необходимость в разработке более экономически эффективных и автономных методов увеличения данных, которые могут использовать существующие внутренние признаки изображений для улучшения процесса настройки подсказок. Мотивация данного исследования заключается в разработке метода, который использует внутренние признаки изображений для улучшения настройки подсказок, минимизируя зависимость от внешних источников знаний и снижая сопутствующие затраты. Авторы предлагают инновационный подход, который включает в себя внутреннее увеличение данных, позволяя модели более эффективно извлекать информацию из доступных изображений и улучшать свою производительность без необходимости в дополнительных внешних данных. Это исследование стремится продемонстрировать, что использование внутренних признаков может существенно повысить качество настройки подсказок, обеспечивая при этом более экономически эффективное и масштабируемое решение для практических приложений. МЕТОДОЛОГИЯ И ПОДХОД Предложенный в статье метод, названный Augmentation-driven Prompt Tuning (AugPT), представляет собой самодостаточный подход к настройке подсказок, основанный на внутреннем увеличении данных. Основная идея заключается в использовании самонаправленного увеличения на необработанных изображениях из обучающего набора данных для более полного использования известных признаков. Это позволяет модели извлекать больше информации из доступных данных, минимизируя зависимость от внешних источников. Технически метод AugPT включает в себя несколько ключевых компонентов. Во-первых, он использует самонаправленное увеличение на необработанных изображениях. Это включает в себя применение различных техник аугментации, таких как случайные обрезки, повороты, изменения яркости и контраста, которые позволяют создать разнообразные представления одного и того же изображения. Эти аугментированные представления затем используются для обучения модели, что позволяет улучшить ее способность к обобщению и адаптации к новым данным. Кроме того, AugPT включает в себя новый механизм фильтрации на основе консенсусного теста, который позволяет автоматически отсеивать шумные образцы из аугментированных данных. Этот механизм использует предварительно обученную модель настройки подсказок в качестве основы для фильтрации, что позволяет модели самостоятельно определять, какие из аугментированных представлений полезны для обучения, а какие следует отбросить. Это способствует повышению качества аугментированных представлений и улучшает общую производительность модели. Архитектурно AugPT строится на основе существующей модели настройки подсказок, что позволяет легко интегрировать его в существующие рабочие процессы без необходимости в значительных изменениях или дополнительных ресурсах. Алгоритмически подход включает в себя итеративный процесс обучения, где модель последовательно обучается на аугментированных данных, фильтруя при этом шумные образцы и обновляя свои параметры на основе высококачественных представлений. Это обеспечивает устойчивое улучшение модели и позволяет ей более эффективно адаптироваться к новым задачам и данным. ЭКСПЕРИМЕНТАЛЬНАЯ ЧАСТЬ Экспериментальная часть исследования направлена на оценку эффективности предложенного метода AugPT в сравнении с традиционными подходами к настройке подсказок. Для этого авторы провели серию экспериментов, использующих различные датасеты и метрики для оценки производительности модели. Экспериментальная установка включала в себя использование нескольких популярных наборов данных для задач классификации изображений и мультидоменных задач, таких как ImageNet, COCO и другие. Эти наборы данных были выбраны из-за их популярности и доступности, что позволяет легко сравнивать результаты с существующими методами. Метрики оценки включали в себя точность классификации, способность модели к обобщению и устойчивость к шуму в данных. Результаты экспериментов показали, что метод AugPT значительно превосходит существующие подходы к настройке подсказок, особенно в условиях ограниченных данных и отсутствия доступа к внешним источникам знаний. В частности, модель, обученная с использованием AugPT, демонстрировала более высокую точность классификации и лучшую способность к обобщению на новых данных. Это подтверждает гипотезу о том, что внутреннее увеличение данных может существенно улучшить производительность моделей без необходимости в дополнительных ресурсах. Сравнение с существующими методами показало, что AugPT обеспечивает более стабильные результаты и меньшую зависимость от качества исходных данных. Это особенно важно в контексте использования моделей в реальных приложениях, где качество данных может варьироваться, и доступ к внешним источникам может быть ограничен. ПРАКТИЧЕСКАЯ ЗНАЧИМОСТЬ Практическая значимость предложенного подхода заключается в его потенциале для широкого применения в различных областях, связанных с обработкой изображений и текста. Во-первых, метод AugPT может быть полезен в ситуациях, где доступ к внешним источникам знаний ограничен или невозможен, например, в условиях конфиденциальности данных или при работе с уникальными датасетами, для которых нет готовых решений. Преимущества предложенного подхода включают в себя снижение затрат на сбор и обработку данных, повышение устойчивости модели к шуму и улучшение ее способности к обобщению. Это делает AugPT привлекательным решением для компаний и исследовательских организаций, стремящихся оптимизировать свои процессы машинного обучения без необходимости в значительных инвестициях в инфраструктуру и ресурсы. Потенциальное влияние на индустрию и науку также значимо. В индустрии AugPT может быть применен в таких областях, как автоматическое распознавание объектов, анализ изображений для медицинских целей, интеллектуальные системы наблюдения и другие. В науке этот метод может способствовать более глубокому пониманию процессов обучения и адаптации моделей, открывая новые возможности для исследований в области искусственного интеллекта. ВЫВОДЫ И ПЕРСПЕКТИВЫ Основные достижения работы заключаются в демонстрации эффективности внутреннего увеличения данных для настройки подсказок, а также в разработке нового механизма фильтрации, который улучшает качество аугментированных представлений. Исследование показало, что использование внутренних признаков изображений может значительно повысить производительность моделей, минимизируя при этом зависимость от внешних источников знаний. Направления будущих исследований могут включать в себя расширение предложенного метода на другие типы данных и задач, а также исследование возможностей интеграции с другими подходами к увеличению данных и настройке моделей. Кроме того, дальнейшие исследования могут быть направлены на оптимизацию алгоритмических решений и снижение вычислительных затрат, связанных с обучением моделей. Ограничения текущего подхода включают в себя потенциальную зависимость от качества исходных данных и необходимость в дальнейшей валидации на более широком спектре задач и наборов данных. Однако, несмотря на эти ограничения, предложенный метод представляет собой значительный шаг вперед в области настройки подсказок и имеет потенциал для широкого применения в различных областях науки и индустрии.

Abstract

For CLIP-based prompt tuning, introducing more data as additional knowledge for enhancing fine-tuning process is proved to be an effective approach. Existing data amplification strategies for prompt tuning typically rely on external knowledge (e.g., large language models or pre-structured knowledge bases), resulting in higher costs for data collection and processing, while generally ignoring further utilization of features in image modality. To address this, we propose Augmentation-driven Prompt Tuning (AugPT), a self-contained distillation-based prompt tuning approach using only internal augmentation on raw dataset to better exploit known features. Specifically, AugPT employs self-supervised augmentation on unlabeled images in the training set, and introduces a novel gating mechanism based on consensus test, reusing the pre-trained prompt tuning backbone model to spontaneously filter noisy samples, further enhancing the quality of augmented views. Extensive experiments validate that AugPT simultaneously enhances model performance and generalization capability without using appended external knowledge. The code of AugPT is available at: https://github.com/JREion/AugPT .

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