Сцена за сценой: Когда нейросети начинают понимать сложные изображения?

Автор: Денис Аветисян


Новое исследование показывает, что генеративные модели, основанные на диффузии, сталкиваются с трудностями при создании сцен с несколькими объектами, особенно когда требуется обобщение на основе новых комбинаций.

🚀 Квантовые новости

Подключайся к потоку квантовых мемов, теорий и откровений из параллельной вселенной.
Только сингулярные инсайты — никакой скуки.

Присоединиться к каналу
Диффузионные модели демонстрируют трудности при генерации композиционных изображений, содержащих несколько объектов, хотя и способны надежно генерировать отдельные объекты; исследование фокусируется на двух режимах - обобщении концепций (когда модель видела каждую концепцию хотя бы раз, но может испытывать трудности из-за дисбаланса данных) и композиционном обобщении (когда требуется генерация новых комбинаций концепций, не встречавшихся в процессе обучения), при этом для оценки точности генерации используется Geneval (Ghosh et al., 2023) и Stable Diffusion 3 (Esser et al., 2024).
Диффузионные модели демонстрируют трудности при генерации композиционных изображений, содержащих несколько объектов, хотя и способны надежно генерировать отдельные объекты; исследование фокусируется на двух режимах — обобщении концепций (когда модель видела каждую концепцию хотя бы раз, но может испытывать трудности из-за дисбаланса данных) и композиционном обобщении (когда требуется генерация новых комбинаций концепций, не встречавшихся в процессе обучения), при этом для оценки точности генерации используется Geneval (Ghosh et al., 2023) и Stable Diffusion 3 (Esser et al., 2024).

Работа посвящена изучению ограничений диффузионных моделей в задаче генерации многообъектных сцен и анализу влияния предвзятости данных и отсутствия композиционного обобщения.

Несмотря на впечатляющие успехи в генерации реалистичных изображений, диффузионные модели демонстрируют неустойчивость при создании сцен с несколькими объектами. В работе ‘When Do Diffusion Models learn to Generate Multiple Objects?’ исследуются причины этого ограничения, с акцентом на роль данных и способность моделей к обобщению композиционных сцен. Авторы выявили, что сложность сцены играет ключевую роль, а обучение счету объектов особенно затруднено при ограниченном объеме данных, при этом способность к композиционному обобщению снижается с увеличением числа скрытых комбинаций объектов. Какие новые подходы к формированию данных и архитектуре моделей необходимы для достижения надежной генерации сложных, многообъектных сцен?


Пределы Воспроизведения: Композиция как Вызов для Диффузионных Моделей

Диффузионные модели, демонстрирующие впечатляющие результаты в создании фотореалистичных изображений, зачастую испытывают трудности при генерации сложных сцен, требующих понимания композиции. Несмотря на способность достоверно воспроизводить отдельные объекты и текстуры, модели сталкиваются с проблемами при организации этих элементов в когерентное и логичное целое. Это проявляется в неспособности корректно отображать пространственные отношения между объектами, а также в сложностях с поддержанием визуальной согласованности в сложных сценах, где требуется учитывать взаимодействие множества элементов. По сути, модели преуспевают в воспроизведении реальности, но испытывают затруднения в её понимании и композиции, что ограничивает их возможности в создании действительно сложных и убедительных визуальных образов.

Несмотря на впечатляющий объем, существующие наборы данных, такие как LAION-2B, зачастую содержат скрытые предвзятости и демонстрируют недостаточный контроль над композиционными элементами изображений. Это приводит к тому, что модели, обученные на этих данных, испытывают трудности с обобщением — способностью создавать правдоподобные изображения, содержащие комбинации объектов и сцен, которые не были явно представлены в процессе обучения. Например, если модель редко видела изображения, сочетающие «кошку, играющую на скрипке в космосе», она с высокой вероятностью сгенерирует нелогичные или искаженные результаты, даже если каждая отдельная концепция («кошка», «скрипка», «космос») ей хорошо знакома. Отсутствие контроля над композицией ограничивает способность моделей к творческому и гибкому синтезу изображений, препятствуя их применению в более сложных и непредсказуемых сценариях.

Ограничения диффузионных моделей особенно ярко проявляются при генерации изображений с «неизвестными композициями» — сочетаниями концепций, которые не встречались в обучающих данных. Исследования показывают, что по мере увеличения количества новых комбинаций объектов и их взаиморасположения, качество генерируемых изображений существенно снижается. Модели, обученные на обширных, но ограниченных наборами данных, испытывают трудности с обобщением и созданием реалистичных сцен, если требуются комбинации, не зафиксированные в процессе обучения. Это указывает на фундаментальную проблему в способности моделей к пониманию и воспроизведению сложных композиционных правил, а не просто к запоминанию существующих визуальных паттернов.

Оценка способности моделей к композиционному синтезу требует специализированных бенчмарков, таких как GenEval. Результаты тестирования демонстрируют, что даже при использовании огромных наборов данных и передовых архитектур, надежное создание многообъектных композиций остается сложной задачей. Модели, обученные на обширных, но недостаточно структурированных данных, часто испытывают трудности при генерации изображений, включающих комбинации объектов и сцен, которые не встречались в процессе обучения. Это указывает на необходимость разработки более эффективных методов обучения и оценки, способных преодолеть ограничения существующих подходов и обеспечить генерацию действительно сложных и разнообразных визуальных композиций. Наблюдается, что производительность моделей резко снижается при увеличении числа композиционных комбинаций, которые не были представлены в обучающем наборе данных.

Матрица показывает, как изменяется доля невидимых комбинаций концептов (оранжевый цвет) при варьировании количества удаленных диагоналей, что позволяет оценить способность диффузионных моделей к обобщению на невидимые комбинации, даже когда отдельные концепты представлены полностью <span class="katex-eq" data-katex-display="false">A_i \times B_j</span>.
Матрица показывает, как изменяется доля невидимых комбинаций концептов (оранжевый цвет) при варьировании количества удаленных диагоналей, что позволяет оценить способность диффузионных моделей к обобщению на невидимые комбинации, даже когда отдельные концепты представлены полностью A_i \times B_j.

Диагностика Проблемы: Контролируемые Композиционные Наборы Данных

Для преодоления ограничений существующих наборов данных, исследователи разработали Mosaic Dataset, предоставляющий точный контроль над атрибутами, количеством объектов и их пространственным расположением. В отличие от традиционных подходов, где эти параметры часто определяются случайным образом или ограничены существующими изображениями, Mosaic позволяет задавать конкретные комбинации атрибутов (например, цвет, форма, размер) и точно контролировать количество экземпляров каждого объекта. Кроме того, набор данных обеспечивает возможность детального управления пространственными отношениями между объектами, такими как «слева от», «над», или «рядом с», что позволяет создавать сцены с заранее определенной сложностью и структурой. Такой подход позволяет проводить целенаправленное исследование способности моделей к обобщению и пониманию композиционных сцен.

Набор данных Mosaic использует схему «Сетчатое расположение» (Grid Layout) для упрощения организации объектов на изображениях. Данная схема предполагает размещение объектов в строго определенных ячейках сетки, что позволяет исследователям контролировать композицию сцен и, как следствие, проводить целенаправленное изучение способности моделей к обобщению при изменении комбинаций атрибутов и отношений между объектами. Такое структурированное расположение упрощает анализ и позволяет изолировать конкретные аспекты композиционного обобщения, такие как понимание атрибутов и пространственных связей, от других факторов, которые могут влиять на производительность модели.

Возможность точного контроля над атрибутами и пространственным расположением объектов в наборах данных, таких как Mosaic, позволяет проводить целенаправленное тестирование навыков атрибуции — корректного соотнесения признаков с объектами — и понимания пространственных взаимосвязей. Исследователи могут создавать сцены, где конкретные атрибуты однозначно связаны с определенными объектами, чтобы оценить способность модели к правильной ассоциации. Контролируемое изменение атрибутов и их пространственного расположения позволяет изолировать и количественно оценить вклад каждого фактора в общую производительность модели при решении задач, связанных с пониманием визуальных сцен.

Набор данных SPEC дополняет Mosaic, обеспечивая дополнительную поддержку оценки производительности в задачах подсчета объектов и анализа сложных пространственных отношений. Эксперименты показывают, что производительность моделей в этих задачах улучшается при тонкой настройке (fine-tuning), однако точность подсчета остается относительно низкой по сравнению с другими навыками. Это указывает на то, что, несмотря на прогресс в понимании сложных пространственных конфигураций, задачи, требующие точного определения количества объектов, представляют собой более сложную проблему для современных моделей компьютерного зрения.

Набор обучающих примеров, созданный методом mosaic (без сетки, RQ1), используется для оценки производительности в задачах подсчета, определения пространственных отношений и связывания атрибутов.
Набор обучающих примеров, созданный методом mosaic (без сетки, RQ1), используется для оценки производительности в задачах подсчета, определения пространственных отношений и связывания атрибутов.

Архитектурные Соображения: За Пределами U-Net

Несмотря на то, что архитектура U-Net продолжает широко использоваться в качестве базового компонента для диффузионных моделей, её внутренние ограничения становятся очевидными при работе со сложными композициями. U-Net, изначально разработанная для сегментации изображений, испытывает трудности при моделировании взаимосвязей между большим количеством объектов и их атрибутами в сложных сценах. Это проявляется в снижении качества генерации при увеличении числа объектов в композиции и в неспособности корректно воссоздавать ранее не встречавшиеся комбинации элементов. Архитектура U-Net, основанная на свёрточных нейронных сетях, имеет ограниченную способность к моделированию глобальных зависимостей, что критично для генерации реалистичных и согласованных композиций.

В последнее время для улучшения композиционного рассуждения в моделях диффузии активно исследуются альтернативы U-Net, в частности, архитектура Diffusion Transformer (DiT). DiT использует трансформаторные архитектуры, позволяющие моделировать взаимосвязи между объектами в сцене более эффективно, чем традиционные сверточные сети. В отличие от U-Net, DiT оперирует последовательностями токенов, представляющих различные части изображения или сцены, что позволяет учитывать глобальный контекст и зависимости между элементами. Это особенно важно при работе со сложными композициями, где необходимо учитывать не только отдельные объекты, но и их взаимное расположение и взаимодействие.

Несмотря на внедрение новых архитектур, таких как Diffusion Transformer (DiT), направленных на повышение устойчивости и обобщающей способности моделей к проблеме дисбаланса данных в обучающих наборах, точность выполнения задач по подсчету объектов остается на уровне приблизительно 44%. Данный показатель указывает на сохраняющиеся трудности в эффективном использовании информации из неполных или смещенных данных, даже при применении передовых архитектурных решений. Неравномерное распределение данных по различным комбинациям объектов и их количеству оказывает существенное влияние на качество обучения и способность модели к обобщению.

Эффективность моделей генерации изображений существенно снижается при увеличении доли не встречавшихся ранее комбинаций объектов в тестовом наборе данных. Наблюдается резкое падение точности, когда более половины комбинаций объектов отсутствуют в обучающей выборке, что указывает на недостаточную способность к обобщению и экстраполяции. Кроме того, точность модели значительно ниже при генерации сцен с большим количеством объектов (например, 10 объектов), составляя примерно 44%, по сравнению с более простыми сценами, где точность достигает около 100%.

Обучение на меньших наборах данных и с увеличением количества неизвестных композиций приводит к резкому снижению производительности Unet в задачах, требующих понимания атрибутов, пространственных связей и подсчета объектов.
Обучение на меньших наборах данных и с увеличением количества неизвестных композиций приводит к резкому снижению производительности Unet в задачах, требующих понимания атрибутов, пространственных связей и подсчета объектов.

Исследование показывает, что диффузионные модели, несмотря на впечатляющие результаты в генерации изображений, сталкиваются с трудностями при создании сложных многообъектных сцен. Данная работа подчеркивает, что производительность моделей снижается по мере увеличения композиционной сложности, что напрямую связано с предвзятостью данных и недостаточной способностью к обобщению. Как отмечает Фэй-Фэй Ли: «Искусственный интеллект должен быть полезен, а не просто сложным». Это высказывание особенно актуально в контексте данной статьи, ведь сложность алгоритма не гарантирует его применимость к реальным задачам, требующим способности к обобщению и преодолению предвзятости данных.

Куда двигаться дальше?

Представленное исследование, хотя и демонстрирует ограниченность диффузионных моделей в генерации сложных многообъектных сцен, поднимает вопросы, требующие более фундаментального пересмотра подхода. Очевидно, что простого увеличения объема данных недостаточно, если модель не способна к композиционному обобщению. Проблема заключается не в недостатке информации, а в неспособности извлечь из неё инвариантные признаки и установить логические связи между объектами. Иначе говоря, модель «видит» пиксели, но не «понимает» сцену.

Дальнейшие исследования должны сосредоточиться на разработке архитектур, способных к явной декомпозиции сцены и моделированию взаимосвязей между объектами. Необходимо отойти от концепции «чёрного ящика» и стремиться к созданию моделей, внутреннюю логику которых можно проанализировать и проверить. Особый интерес представляет формализация понятия «композиционности» и разработка метрик для оценки способности моделей к обобщению на новые комбинации объектов. В противном случае, мы обречены на создание систем, способных лишь имитировать разум, но не обладающих подлинным пониманием.

В конечном итоге, истинный прогресс в данной области потребует интеграции достижений в области символьного искусственного интеллекта и глубокого обучения. Необходимо создать системы, способные сочетать мощь нейронных сетей с точностью логических рассуждений. Иначе говоря, нужно построить не просто генератор изображений, а полноценного «видящего» агента, способного к осмысленному взаимодействию с миром.


Оригинал статьи: https://arxiv.org/pdf/2605.00273.pdf

Связаться с автором: https://www.linkedin.com/in/avetisyan/

Смотрите также:

2026-05-05 05:34