Verifica Formale di Surrogati Neurali per PDE tramite Compilazione SMT
I ricercatori hanno sviluppato un metodo per verificare formalmente gli Operatori Neurali di Fourier (FNO) per la simulazione di PDE utilizzando risolutori SMT. Sfruttando la struttura lineare a tratti degli FNO dopo l'addestramento, compilano la convoluzione spettrale in aritmetica lineare reale in Z3. Vengono proposte due codifiche: una codifica esatta, valida per dimostrazioni e controesempi, e una codifica congelata più veloce che approssima il percorso spettrale. Su 10 piccoli surrogati FNO per il problema 1D di avvezione-diffusione-reazione, la codifica esatta ha prodotto 2 dimostrazioni valide di positività su modelli lineari, 5 controesempi validi di positività e 10 controesempi validi di violazione della massa. Questo lavoro affronta la mancanza di garanzie formali nei surrogati neurali per PDE, consentendo la verifica di proprietà fisiche come la positività e la conservazione della massa.
Fatti principali
- Gli Operatori Neurali di Fourier (FNO) sono utilizzati per accelerare la simulazione di PDE.
- La convoluzione spettrale in un FNO è una mappa lineare dopo che i pesi addestrati e la griglia sono fissati.
- Il forward pass di un FNO è lineare a tratti e può essere rappresentato nell'aritmetica lineare reale di Z3.
- Vengono studiate due codifiche: codifica esatta (moltiplicazione di matrici dense) e codifica congelata (percorso spettrale costante).
- Gli esperimenti sono stati condotti su 10 piccoli surrogati FNO per il problema 1D di avvezione-diffusione-reazione.
- I modelli avevano da 85 a 117 parametri e griglie di dimensione da 8 a 32.
- La codifica esatta ha fornito 2 dimostrazioni valide di positività su modelli lineari (senza ReLU).
- La codifica esatta ha fornito 5 controesempi validi di positività e 10 controesempi validi di violazione della massa.
- Le restanti 3 dimostrazioni di positività non erano valide (probabilmente a causa della non linearità ReLU).
- Il lavoro mira a fornire garanzie formali per i surrogati neurali di PDE.
Entità
—