Lifted Schrödinger Bridges for Gaussian Mixture Endpoints: Projection Gaps and Path-Space Obstructions

Resumen Técnico: Puentes de Schrödinger Elevados para Puntas de Mezclas Gaussianas

Enunciado del Problema
El artículo aborda el problema de control de densidad estocástica (puente de Schrödinger) entre dos distribuciones de punto final, $\rho_0$ y $\rho_1$, especificadas como Modelos de Mezcla Gaussiana (GMM). Si bien el puente de Schrödinger entre puntos finales puramente gaussianos admite una solución de forma cerrada bajo priores de movimiento browniano, el puente entre GMMs generales no permanece dentro de la clase de mezclas gaussianas finitas con dinámicas de forma cerrada. La dificultad central radica en la ambigüedad estructural: el proceso de estado no porta etiquetas que indiquen qué componente de mezcla inicial debe emparejarse con qué componente terminal. Resolver directamente la deriva óptima en el espacio de trayectorias sin etiquetas es computacionalmente intratable para altas dimensiones y carece de una forma analítica explícita.

Metodología: La Construcción del Espacio de Trayectorias Elevado
Los autores proponen una formulación "elevada" que aumenta el espacio de trayectorias con etiquetas discretas auxiliares que representan los componentes de mezcla de origen y destino.

1. Aumento: El espacio de estados se eleva desde el espacio de trayectorias $\Omega$ a $\bar{\Omega} = \mathcal{Z} \times \Omega$, donde $\mathcal{Z} = \mathcal{I} \times \mathcal{J}$ representa pares de índices de componentes inicial ($i$) y terminal ($j$).
2. Descomposición: En este espacio aumentado, el problema se descompone en:
   • Puentes Gaussianos Pareados: Para cada par fijo $(i, j)$, el problema se reduce a un puente de Schrödinger gaussiano estándar entre los componentes específicos $p^0_i$ y $p^1_j$. Estos admiten soluciones explícitas de forma cerrada para la densidad marginal $\rho^t_{ij}$, la deriva óptima $u^t_{ij}$ y el costo de energía cinética $C_{ij}$.
   • Acoplamiento de Dimensión Finita: La asignación de masa entre componentes está gobernada por una matriz de acoplamiento $\pi \in \Pi(\alpha^0, \alpha^1)$, que minimiza un objetivo de dimensión finita que combina los costos pareados y un término de regularización entrópico relativo a un acoplamiento previo $\eta$.
3. Optimización: El acoplamiento óptimo $\pi^*$ se encuentra mediante escalado de Sinkhorn, equilibrando los costos de transporte $C_{ij}$ contra la entropía de la asignación.
4. Proyección: La solución en el espacio elevado se proyecta de vuelta al espacio de trayectorias sin etiquetas original "olvidando" las etiquetas. Esto produce un flujo de densidad solo de estado factible $\rho^\pi_t$ y una deriva de retroalimentación de Markov promediada a posteriori $\bar{u}^\pi_t$.

Contribuciones Clave y Resultados Teóricos

• Reducción Elevada Exacta (Teorema 1): Los autores demuestran que minimizar la entropía relativa en el espacio etiquetado elevado es equivalente a un problema de transporte óptimo entrópico de dimensión finita sobre los pesos de la mezcla. La ley elevada óptima es una mezcla de puentes gaussianos pareados ponderados por el acoplamiento óptimo $\pi^*$.
• Identidad de la Brecha de Proyección (Teorema 2): Una contribución teórica central es la derivación de una identidad que cuantifica la discrepancia entre la solución elevada y el puente de Schrödinar sin etiquetas directo. La entropía relativa de la ley elevada se descompone en la entropía relativa de la ley proyectada (sin etiquetas) más una brecha de información de etiqueta condicional no negativa, $G(\pi, \eta)$.
  • Esta brecha representa una "obstrucción en el espacio de trayectorias": el optimizador elevado, al proyectarse, generalmente no es el verdadero optimizador del problema sin etiquetas original porque la ley proyectada retiene una memoria oculta de las etiquetas de los componentes a través de la trayectoria, mientras que el verdadero puente sin etiquetas no lo hace.
  • El artículo identifica una condición específica de "potencial común de trayectoria" (Proposición 5) bajo la cual esta brecha desaparece, mostrando que la obstrucción no es inherente a la elevación sino que depende de la estructura de las derivas.
• Proyección Markoviana y Factibilidad (Proposiciones 6 y 7): El artículo construye una deriva de retroalimentación de estado $\bar{u}^\pi_t$ promediando a posteriori las derivas pareadas. Se demuestra que:
  • Esta deriva es regular (localmente Lipschitz, crecimiento lineal) y satisface la ecuación de Fokker-Planck, asegurando que se cumplan los puntos finales.
  • La ley proyectada $Q_\pi$ (obtenida olvidando las etiquetas) es generalmente no markoviana. La deriva construida $\bar{u}^\pi$ es la proyección ortogonal $L^2$ de la verdadera deriva de etiqueta oculta (no markoviana) sobre el espacio de controles de retroalimentación de estado.
• Descomposición de Entropía (Teorema 3): El costo total se descompone en tres niveles: el costo de la retroalimentación de Markov verdadera, el costo de la markovización (aproximando la ley de trayectoria no markoviana con una markoviana) y la brecha de proyección (pérdida de información por descartar las etiquetas).

Resultados Numéricos
El artículo valida el enfoque mediante varios experimentos numéricos:

• Dirigido de GMMs 1D y 2D: El método dirige con éxito distribuciones multimodales complejas (por ejemplo, dividir una sola gaussiana en ocho modos, o transformar una forma de media luna en una estrella).
• Eficiencia: Comparado con solucionadores de puente de Schrödinger basados en cuadrícula directa, el enfoque elevado es significativamente más rápido (órdenes de magnitud en 2D) y escala con el número de componentes de mezcla ($N_1 \times N_2$) en lugar del tamaño de la cuadrícula ($M^{2d}$).
• Rendimiento: La energía cinética de la retroalimentación de Markov proyectada es ligeramente superior al mínimo teórico del puente sin etiquetas directo (por ejemplo, una brecha de ~3.3% en un ejemplo 1D), consistente con la brecha de proyección teórica.
• Sensibilidad al Prior: Los experimentos demuestran que la elección del acoplamiento previo $\eta$ (por ejemplo, independiente vs. favorecedor de la diagonal) influye directamente en la asignación de componentes resultante $\pi^*$, actuando como una entrada de modelado.

Significado y Afirmaciones
El artículo afirma que su contribución principal es un marco conceptual y computacional que reinterpreta los puentes de Schrödinger de mezcla gaussiana como un problema de espacio de trayectorias elevado.

• Proporciona una solución explícita, estructurada e interpretable que evita la "maldición de la dimensionalidad" asociada con los métodos basados en cuadrícula.
• Separa rigurosamente tres conceptos distintos a menudo confundidos en el modelado generativo: la optimalidad en el espacio etiquetado aumentado, la factibilidad de la proyección al espacio de estados y la brecha de información que impide que la ley proyectada sea el óptimo sin etiquetas exacto.
• Los autores enfatizan que, aunque la retroalimentación de Markov proyectada no es la solución exacta al problema sin etiquetas original, proporciona una aproximación certificada, de energía finita y computacionalmente tratable que es numéricamente cercana a la solución directa, lo que la hace adecuada para tareas prácticas de control de densidad y modelado generativo que involucran distribuciones multimodales.