Просмотр исходного текста страницы Генеративные модели на основе диффузии

== Общая информация ==
Курс предназначен для студентов 4 курса ФКН ПМИ, но приглашаются все желающие, уверенно знающие математику младших курсов (в особенности теорию вероятностей), базово ориентирующиеся в глубинном обучении и программировании на PyTorch.

Занятия проходят по субботам в 13:00-16:00, аудитория будет сообщаться в чате.

Лектор/семинарист: [https://t.me/denrakitin23 Денис Ракитин]

[https://t.me/+WidqaBPBslQ1MGIy Чат курса]

[https://disk.yandex.ru/i/uBUdD8VPm36iAQ Программа и описание курса]

[https://github.com/RakitinDen/ODE-SDE-Generative-Models Старый туториал по ODE/SDE моделям, близкий к программе курса]

== Оценки ==

=== [https://docs.google.com/spreadsheets/d/1aONSCqla2-VBE7kWVrAqaDS7rwqFubs57O24Kffa3yE/edit?usp=sharing Таблица с оценками] ===
Формула итоговой оценки: О<sub>итог</sub> = 0.5 * О<sub>дз</sub> + 0.2 * О<sub>проект</sub> + 0.3 * О<sub>экз</sub>. Округление арифметическое.

== Домашние задания ==

Сдаются в [https://classroom.google.com/c/NzE3ODc5ODg4NTUw?cjc=mzafi45 классрум]. Код курса '''mzafi45'''.

В курсе будут 4 теоретических и 2 практических домашних задания. Вес у всех задач одинаковый, задачи оцениваются из 1 балла. Если в задачах есть пункты, то будет подписано количество баллов за каждый пункт, иначе баллы между пунктами делятся поровну.

===Теория===

[https://disk.yandex.ru/i/Uv4wH9ScwdOtPQ ДЗ-1], [https://www.overleaf.com/read/mxycqmzbfrgj#8cbb38 TeX исходник], дедлайн 6 октября 23:59.

[https://disk.yandex.ru/i/vony06J3moTo1g ДЗ-2], [https://www.overleaf.com/read/fxfxrgbzyrht#6113ed ТеХ исходник], дедлайн 3 ноября 23:59.

[https://disk.yandex.ru/i/-YKGsduwV0X55g ДЗ-3], [https://www.overleaf.com/read/dxjtvqcxtcqw#5af7ca ТеХ исходник], дедлайн 24 ноября 23:59.

===Практика===

[https://disk.yandex.ru/d/LU6aLw4tUTdnfQ ДЗ-1], дедлайн 10 ноября 23:59.

== Проект ==

Состоит в реализации и проведении экспериментов с одной из рассмотренных на курсе моделей. 

== Лекции и семинары ==
[https://disk.yandex.ru/d/xGW19AWc148l0Q Записи лекций и семинаров]

[https://disk.yandex.ru/d/XLco2cbeCzBwQQ Семинарские задачи]

'''Лекция 1.''' Генеративное моделирование. Семейства генеративных моделей: вариационные автокодировщики (VAEs), генеративно-состязательные сети (GANs), диффузионные модели. Генеративная трилемма: генеративная модель должна обладать высоким качеством генерации, высоким разнообразием и скоростью генерирования. Повтор теории вероятностей: совместная и условная плотность, формула Байеса, подсчет матожиданий через плотности. Условное матожидание (УМО): определение через интеграл условной плотности. Свойства: линейность, формула полного матожидания, вынос функции от условия за УМО, УМО от независимой величины равна безусловному матожиданию. Условное матожидание как наилучшее предсказание в среднеквадратичном.

'''Семинар 1.''' Гауссовские векторы/многомерное нормальное распределение. Три эквивалентных определения, формула плотности. Аффинная замена гауссовских векторов, эквивалентность независимости и некоррелированности их компонент, ортогональное разложение, условное распределение компоненты относительно другой компоненты.

'''Лекция 2.''' Процессы зашумления с дискретным временем: процесс с сохраняющейся (variance preserving, VP) и взрывающейся (variance exploding, VE) дисперсией. Определение марковской цепи. Префиксные суммы независимых величин образуют марковскую цепь. Процессы зашумления являются марковскими цепями. Работа с марковскими цепями: обуславливание марковской цепи на начальное состояние эквивалентно построению марковской цепи, стартующей из этого состояния. Переходные плотности VP и VE процессов. Эвристичная схема обращения процесса зашумления по времени [https://arxiv.org/abs/2006.11239]. Обучение нейросети-денойзера для предсказания чистой картинки по шумной.

'''Семинар 2.''' Обращенная по времени марковская цепь является марковской цепью. Обыкновенные дифференциальные уравнения (ОДУ), схема Эйлера. Построение непрерывного аналога процесса удаления картинки, заданного через ОДУ. Определение Винеровского процесса. (Псевдо)-определение стохастических дифференциальных уравнений (СДУ) через схему Эйлера. Построение непрерывных по времени аналогов VP и VE процессов: СДУ с сохраняющейся (VP-SDE) и взрывающейся (VE-SDE) дисперсией. Подсчет переходных распределений VE-SDE и VP-SDE с помощью решения линейных ОДУ.

'''Лекция 3.''' Эволюция плотности частицы, движущейся под действием ОДУ/СДУ: уравнение непрерывности и уравнение Фоккера-Планка. Неформальное утверждение о единственности решения уравнения Фоккера-Планка с начальным условием при выполнении условий регулярности.

'''Семинар 3.''' Вывод общего вида законов сохранения (взято из курса [http://wiki.cs.hse.ru/%D0%A3%D1%80%D0%B0%D0%B2%D0%BD%D0%B5%D0%BD%D0%B8%D1%8F_%D1%81_%D1%87%D0%B0%D1%81%D1%82%D0%BD%D1%8B%D0%BC%D0%B8_%D0%BF%D1%80%D0%BE%D0%B8%D0%B7%D0%B2%D0%BE%D0%B4%D0%BD%D1%8B%D0%BC%D0%B8_(2022-2023) УрЧП]). Интерпретация уравнения непрерывности, теплопроводности и Фоккера-Планка через законы сохранения. Решение транспортного уравнения с константным сносом. Дорешивание семинара 2: VP-SDE получается шкалированием VE-SDE.

'''Лекция 4.''' Диффузионные модели с непрерывным временем [https://arxiv.org/abs/2011.13456]. Переход от СДУ к ОДУ и обратно. Обращение ОДУ по времени. Обращение СДУ по времени [https://core.ac.uk/download/pdf/82826666.pdf] с помощью схемы СДУ -> ОДУ -> обратное ОДУ -> обратное СДУ. Score функция, ее представление через условное матожидание условной score функции, интерпретация и обучение методом Denoising Score Matching [https://arxiv.org/abs/1907.05600]. Эквивалентность Denoising Score Matching обучению модели, предсказывающей чистое изображение/шум в изображении.

'''Семинар 4.''' Семплирование из диффузионных моделей с непрерывным временем: дискретизация обратного СДУ по схеме Эйлера. Связь с дискретной схемой обращения процесса по времени из лекции 2. Выражение score функции VP-SDE через score функцию VE-SDE. Применение: имея обученную модель-денойзер, можно семплировать из VP-SDE и VE-SDE без переобучения модели. Предельное распределение однородного СДУ, связь со стационарным распределением, сведение к однородному уравнению Фоккера-Планка (d/dt p_t(x) = 0). Динамика Ланжевена: снос в СДУ, сходящейся к распределению p(x), с точностью до константы совпадает со score функцией. Применение к генерации семплов из распределения p(x) и связь с градиентным подъемом.

'''Лекция / Семинар 5.''' Имплементация диффузионных моделей на практике: [https://disk.yandex.ru/d/wu1nZFTnW5SWpg ноутбук]. Denoising Score Matching для обучения обусловленных диффузионных моделей.

'''Лекция / Семинар 6.''' Обуславливание безусловных диффузионных моделей с помощью шумного классификатора: [https://arxiv.org/abs/2105.05233 Classifier Guidance], контролирование баланса между правдоподобием и разнообразием семплов. Перенос той же техники на предобученные условные модели: [https://arxiv.org/abs/2207.12598 Classifier-free Guidance]. Дистилляция диффузионных моделей. Первое семейство моделей: обучение генератора на повтор траектории ОДУ диффузионной модели: [https://arxiv.org/abs/2202.00512 Progressive Distillation], [https://arxiv.org/abs/2303.01469 Consistency Models], [https://arxiv.org/abs/2306.05544 BOOT]. Использование диффузионной модели в качестве обучающего сигнала для произвольного вида генератора: метод [https://arxiv.org/abs/2311.18828 Distribution Matching Distillation] (его прототип [https://arxiv.org/abs/2305.18455 Diff-Instruct], конкурентная работа [https://arxiv.org/abs/2312.05239 SwiftBrush] и улучшенная версия [https://arxiv.org/html/2405.14867v1 Improved DMD]).

'''Лекция / Семинар 7.''' Диффузионные модели как процессы интерполяции между шумом и данными. Модель [https://arxiv.org/abs/2210.02747 Flow Matching] (аналоги [https://arxiv.org/abs/2302.00482 Conditional Flow Matching] и [https://arxiv.org/abs/2209.15571 Stochastic Interpolants]): построение ОДУ, порождающего безусловную динамику при известном ОДУ, порождающем условную динамику. Применение к задаче безусловной генерации. Применение к парным задачам переноса стиля: Flow Matching [https://arxiv.org/abs/2209.03003 снижает] транспортную цену. Представление диффузионных моделей в качестве частного случая модели Flow Matching. Статьи про решение парных задач [https://arxiv.org/abs/2310.03725 на основе ОДУ] и [https://arxiv.org/abs/2302.05872 на основе СДУ].

== Материалы прошлых лет ==

[http://wiki.cs.hse.ru/%D0%93%D0%B5%D0%BD%D0%B5%D1%80%D0%B0%D1%82%D0%B8%D0%B2%D0%BD%D1%8B%D0%B5_%D0%BC%D0%BE%D0%B4%D0%B5%D0%BB%D0%B8_%D0%BD%D0%B0_%D0%BE%D1%81%D0%BD%D0%BE%D0%B2%D0%B5_ODE_%D0%B8_SDE Вики-страница]

[https://disk.yandex.ru/d/BMKdeMsSPKW4VQ Видеозаписи лекций]