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

Riepilogo Tecnico: Ponti di Schrödinger Sollevati per Estremi a Miscele Gaussiane

Enunciato del Problema
Il lavoro affronta il problema del controllo stocastico della densità (ponte di Schrödinger) tra due distribuzioni agli estremi, $\rho_0$ e $\rho_1$, specificate come Modelli a Miscele Gaussiane (GMM). Mentre il ponte di Schrödinger tra estremi puramente gaussiani ammette una soluzione in forma chiusa sotto priori browniani, il ponte tra GMM generali non rimane all'interno della classe delle miscele gaussiane finite con dinamiche in forma chiusa. La difficoltà fondamentale risiede nell'ambiguità strutturale: il processo di stato non porta etichette che indicano quale componente iniziale della miscela debba essere accoppiata a quale componente terminale. Risolvere direttamente la deriva ottimale sullo spazio dei percorsi non etichettati è computazionalmente intrattabile per alte dimensioni e manca di una forma analitica esplicita.

Metodologia: La Costruzione dello Spazio dei Percorsi Sollevato
Gli autori propongono una formulazione "sollevata" che arricchisce lo spazio dei percorsi con etichette discrete ausiliarie che rappresentano le componenti di origine e destinazione della miscela.

1. Arricchimento: Lo spazio degli stati è sollevato dallo spazio delle traiettorie $\Omega$ a $\bar{\Omega} = \mathcal{Z} \times \Omega$, dove $\mathcal{Z} = \mathcal{I} \times \mathcal{J}$ rappresenta le coppie di indici delle componenti iniziali ($i$) e terminali ($j$).
2. Decomposizione: Su questo spazio arricchito, il problema si decompone in:
   • Ponti Gaussiani a Coppie: Per ogni coppia fissa $(i, j)$, il problema si riduce a un ponte di Schrödinger gaussiano standard tra le specifiche componenti $p^0_i$ e $p^1_j$. Queste ammettono soluzioni esplicite in forma chiusa per la densità marginale $\rho^t_{ij}$, la deriva ottimale $u^t_{ij}$ e il costo dell'energia cinetica $C_{ij}$.
   • Accoppiamento a Dimensione Finita: L'assegnazione della massa tra le componenti è governata da una matrice di accoppiamento $\pi \in \Pi(\alpha^0, \alpha^1)$, che minimizza un obiettivo a dimensione finita combinando i costi a coppie e un termine di regolarizzazione entropica rispetto a un accoppiamento priori $\eta$.
3. Ottimizzazione: L'accoppiamento ottimale $\pi^*$ è trovato tramite scalatura di Sinkhorn, bilanciando i costi di trasporto $C_{ij}$ contro l'entropia dell'assegnazione.
4. Proiezione: La soluzione sullo spazio sollevato è proiettata indietro sullo spazio dei percorsi non etichettati originale "dimenticando" le etichette. Ciò produce un flusso di densità fattibile solo nello stato $\rho^\pi_t$ e una deriva di feedback markoviana media a posteriori $\bar{u}^\pi_t$.

Contributi Chiave e Risultati Teorici

• Riduzione Esatta Sollevata (Teorema 1): Gli autori dimostrano che la minimizzazione dell'entropia relativa sullo spazio etichettato sollevato è equivalente a un problema di trasporto ottimo entropico a dimensione finita sui pesi della miscela. La legge sollevata ottimale è una miscela di ponti gaussiani a coppie ponderata dall'accoppiamento ottimale $\pi^*$.
• Identità del Divario di Proiezione (Teorema 2): Un contributo teorico centrale è la derivazione di un'identità che quantifica la discrepanza tra la soluzione sollevata e il ponte di Schrödinger non etichettato diretto. L'entropia relativa della legge sollevata si decompone nell'entropia relativa della legge proiettata (non etichettata) più un divario informativo condizionale alle etichette non negativo, $G(\pi, \eta)$.
  • Questo divario rappresenta un "ostacolo nello spazio dei percorsi": l'ottimizzatore sollevato, quando proiettato, generalmente non è il vero ottimizzatore del problema non etichettato originale, poiché la legge proiettata trattiene una memoria nascosta delle etichette delle componenti attraverso la traiettoria, mentre il vero ponte non etichettato non lo fa.
  • Il lavoro identifica una specifica condizione di "potenziale comune dei percorsi" (Proposizione 5) sotto la quale questo divario svanisce, mostrando che l'ostacolo non è intrinseco al sollevamento ma dipende dalla struttura delle derive.
• Proiezione Markoviana e Fattibilità (Proposizioni 6 e 7): Il lavoro costruisce una deriva di feedback sullo stato $\bar{u}^\pi_t$ mediante media a posteriori delle derive a coppie. È dimostrato che:
  • Questa deriva è regolare (localmente lipschitziana, crescita lineare) e soddisfa l'equazione di Fokker-Planck, garantendo che gli estremi siano raggiunti.
  • La legge proiettata $Q_\pi$ (ottenuta dimenticando le etichette) è generalmente non markoviana. La deriva costruita $\bar{u}^\pi$ è la proiezione ortogonale $L^2$ della vera deriva con etichette nascoste (non markoviana) sullo spazio dei controlli di feedback sullo stato.
• Decomposizione Entropica (Teorema 3): Il costo totale è decomposto in tre livelli: il costo della vera feedback markoviana, il costo della markovianizzazione (approssimazione della legge di percorso non markoviana con una markoviana) e il divario di proiezione (perdita di informazione derivante dallo scarto delle etichette).

Risultati Numerici
Il lavoro convalida l'approccio attraverso diversi esperimenti numerici:

• Guida di GMM 1D e 2D: Il metodo guida con successo distribuzioni complesse multimodali (ad esempio, dividendo una singola gaussiana in otto modi, o trasformando una forma a mezzaluna in una stella).
• Efficienza: Rispetto ai solver diretti di ponti di Schrödinger basati su griglia, l'approccio sollevato è significativamente più veloce (ordini di grandezza in 2D) e scala con il numero di componenti della miscela ($N_1 \times N_2$) piuttosto che con la dimensione della griglia ($M^{2d}$).
• Prestazioni: L'energia cinetica della feedback markoviana proiettata è leggermente superiore al minimo teorico del ponte non etichettato diretto (ad esempio, un divario di circa il 3,3% in un esempio 1D), coerente con il divario di proiezione teorico.
• Sensibilità al Prior: Gli esperimenti dimostrano che la scelta dell'accoppiamento priori $\eta$ (ad esempio, indipendente vs. favorevole alla diagonale) influenza direttamente l'assegnazione delle componenti risultante $\pi^*$, agendo come un input di modellazione.

Significato e Affermazioni
Il lavoro afferma che il suo contributo principale è un quadro concettuale e computazionale che riformula i ponti di Schrödinger a miscele gaussiane come un problema di spazio dei percorsi sollevato.

• Fornisce una soluzione esplicita, strutturata e interpretabile che evita la "maledizione della dimensionalità" associata ai metodi basati su griglia.
• Separa rigorosamente tre concetti distinti spesso confusi nella modellazione generativa: l'ottimalità nello spazio etichettato arricchito, la fattibilità della proiezione allo spazio degli stati e il divario informativo che impedisce alla legge proiettata di essere l'ottimo non etichettato esatto.
• Gli autori sottolineano che, sebbene la feedback markoviana proiettata non sia la soluzione esatta al problema non etichettato originale, fornisce un'approssimazione certificata, a energia finita e computazionalmente trattabile che è numericamente vicina alla soluzione diretta, rendendola adatta a compiti pratici di controllo della densità e modellazione generativa che coinvolgono distribuzioni multimodali.