Fractal Flow: Hierarchical and Interpretable Normalizing Flow via Topic Modeling and Recursive Strategy
2508.19750v1
stat.ML, cs.LG
2025-08-30
Авторы:
Binhui Zhang, Jianwei Ma
Резюме на русском
## Контекст
Normalizing Flows (NF) — это мощный фреймворк для высокоразмерной оценки распределений и генерирования данных. Они построены на основе инвертируемых преобразований, позволяющих эффективно оценивать логарифмический детерминант якобиана. Однако, существуют проблемы с эксплицитностью интерпретации трансформаций и улучшением структурированности латентного пространства. Например, стандартные NF могут быть недостаточно точными в задачах кластеризации или генерирования структурированных данных. Эти главные вызовы мотивируют развитие более гибких и интерпретируемых NF-моделей, которые могут эффективно работать в задачах сложных генерируемых процессов.
## Метод
Fractal Flow предлагает два ключевых инновационных решения. Во-первых, интеграция Kolmogorov-Arnold Networks (KAN) с Latent Dirichlet Allocation (LDA) позволяет строить структурированное и интерпретируемое латентное пространство. Это решение позволяет раскладывать латентные представления на семантические кластеры, улучшая их понимание и управляемость. Во-вторых, воспроизводимый модульный дизайн вдохновленный Fractal Generative Models добавляет уровни рекурсии в трансформации NF, увеличивая их выразительность и точность оценки. Архитектура Fractal Flow объединяет эти элементы в инвариантную к инверсии и вычислительно эффективную структуру.
## Результаты
Эксперименты проводились на MNIST, FashionMNIST, CIFAR-10 и данных геофизического происхождения. Модель показала высокую точность оценки распределений и выдавала структурированные латентные пространства. На MNIST и FashionMNIST, Fractal Flow демонстрировал значительную улучшенную точность кластеризации в сравнении с базовыми NF. На CIFAR-10, модель показала высокую точность генерирования и интерпретируемость латентных пространств. На геофизических данных, Fractal Flow существенно повысил точность в задачах задач восстановления и кластеризации.
## Значимость
Fractal Flow может применяться в сложных задачах генерируемых процессов, таких как генерирование структурированных данных, анализ изображений и текстов, в том числе в приложениях к геофизическим процессам. Улучшенная интерпретируемость позволяет ускорить разработку систем, а структурированность латентного пространства облегчает управляемость и эффективность моделей.
## Выводы
На основе предложенных результатов, Fractal Flow доказывает свою эффективность в сравнении с существующими NF-моделями. Будущие исследования будут сфокусированы на расширении модели для более сложных латентных пространств и приложениях в реальных задачах, таких как видео-генерирование и синтез данных в области здравоохранения
Abstract
Normalizing Flows provide a principled framework for high-dimensional density
estimation and generative modeling by constructing invertible transformations
with tractable Jacobian determinants. We propose Fractal Flow, a novel
normalizing flow architecture that enhances both expressiveness and
interpretability through two key innovations. First, we integrate
Kolmogorov-Arnold Networks and incorporate Latent Dirichlet Allocation into
normalizing flows to construct a structured, interpretable latent space and
model hierarchical semantic clusters. Second, inspired by Fractal Generative
Models, we introduce a recursive modular design into normalizing flows to
improve transformation interpretability and estimation accuracy. Experiments on
MNIST, FashionMNIST, CIFAR-10, and geophysical data demonstrate that the
Fractal Flow achieves latent clustering, controllable generation, and superior
estimation accuracy.
Ссылки и действия
Дополнительные ресурсы: