Certificare la robustezza quando cambia il routing del modello
In parole semplici
Alcuni modelli linguistici usano parti esperte diverse a seconda dell’input. Se un certificato di robustezza presume che vengano usate sempre le stesse parti, una piccola modifica al testo potrebbe far sì che il certificato descriva calcoli diversi da quelli realmente eseguiti.
CertMoE verifica che gli input modificati restino entro un insieme più ampio di esperti ammessi, poi controlla che le previsioni restino stabili in un modello limitato a quell’insieme. In un test su un modello per l’analisi del sentiment, aggiungere CertMoE ha ridotto la quota di input certificati in cui gli attacchi hanno trovato cambiamenti nelle previsioni, anche se gli attacchi hanno esplorato solo una parte delle modifiche ammesse.
La ricerca
Un certificato di robustezza dovrebbe indicare che una previsione resta valida anche dopo le modifiche al testo ammesse. Ma in un modello mixture of experts, quelle modifiche possono cambiare gli esperti attivati e quindi i calcoli eseguiti.
La domanda concreta è se il certificato continui a descrivere il modello quando cambia il suo percorso.
CertMoE certifica sia quali esperti possono usare gli input perturbati, sia se le previsioni restano stabili entro quell’insieme certificato.
Un certificato di robustezza è utile solo se copre i calcoli che il modello esegue davvero.
Nei modelli linguistici mixture of experts, un input modificato può selezionare esperti diversi: un metodo di certificazione può tenere conto di questi cambiamenti e verificare comunque che le previsioni restino stabili?
CertMoE certifica prima che gli input perturbati restino entro un insieme più ampio di esperti, poi certifica la stabilità delle previsioni in un modello limitato a quell’insieme. Nel test di classificazione del sentiment di un modello, aggiungere CertMoE a RanMASK ha ridotto dal 32.98% al 6.25% la quota di input certificati in cui gli attacchi hanno trovato cambiamenti nelle previsioni.
Il risultato sostiene l’utilità di certificati più selettivi in quel contesto, ma gli attacchi hanno esplorato solo una parte di ciascuna regione certificata.
Questi risultati degli attacchi derivano dall’esplorazione di una parte soltanto di ciascuna regione certificata: non dimostrano in modo esaustivo che ogni certificato sia valido o non valido.
La valutazione riguarda la classificazione di testo con tre modelli MoE rappresentativi; altri tipi di architettura e le attività di generazione non sono stati testati.
Analisi
Dettagli
- Autori del paper originale
- Qingyue Wang, Yudong Gao, Lixia Wang, Ruixuan Huang, Yuanyuan Yuan
- Inviato ad arXiv
- Licenza
- arXiv non-exclusive