LAMBO: аугментация лендмарков путем семплирования барицентров на многообразии

В статье предлагается новый метод аугментации данных основанных на теории оптимального транспорта, при котором сгенерированные данные гарантированно лежат на исходном многообразии данных. Предложенный алгоритм вычисляет новую синтетическую точку путем выбира клики в графе ближайших соседей построенному по данным и вычисления барицентра Вассерштейна относительно нее с барицентрическими координатами взятыми случайно из равномерного распределения. Будучи чрезвычайно естественной на вид, совокупность подобных барицентров используется для увеличения исходной выборки данных. Метод применен к задаче распознавания лендмарков в изображениях в подходах как без учителя, так и с частичным его привлечением в популярных задачах извлечения ключевых точек лица, определения позы, а также сегментации анатомических контуров в анализе медицинских изображений. Показано что подход с семплированием с барицентров превосходит другие современные методы аугментации данных. Код доступен по ссылке https://github.com/cviaai/LAMBO/