Saved in:
Bibliographic Details
Main Authors: Dang, Chenhao, Ma, Jing
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2511.07888
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908798867734528
author Dang, Chenhao
Ma, Jing
author_facet Dang, Chenhao
Ma, Jing
contents A persistent challenge in text classification (TC) is that enhancing model robustness against adversarial attacks typically degrades performance on clean data. We argue that this challenge can be resolved by modeling the distribution of clean samples in the encoder embedding manifold. To this end, we propose the Manifold-Correcting Causal Flow (MC^2F), a two-module system that operates directly on sentence embeddings. A Stratified Riemannian Continuous Normalizing Flow (SR-CNF) learns the density of the clean data manifold. It identifies out-of-distribution embeddings, which are then corrected by a Geodesic Purification Solver. This solver projects adversarial points back onto the learned manifold via the shortest path, restoring a clean, semantically coherent representation. We conducted extensive evaluations on text classification (TC) across three datasets and multiple adversarial attacks. The results demonstrate that our method, MC^2F, not only establishes a new state-of-the-art in adversarial robustness but also fully preserves performance on clean data, even yielding modest gains in accuracy.
format Preprint
id arxiv_https___arxiv_org_abs_2511_07888
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Breaking the Adversarial Robustness-Performance Trade-off in Text Classification via Manifold Purification
Dang, Chenhao
Ma, Jing
Computation and Language
A persistent challenge in text classification (TC) is that enhancing model robustness against adversarial attacks typically degrades performance on clean data. We argue that this challenge can be resolved by modeling the distribution of clean samples in the encoder embedding manifold. To this end, we propose the Manifold-Correcting Causal Flow (MC^2F), a two-module system that operates directly on sentence embeddings. A Stratified Riemannian Continuous Normalizing Flow (SR-CNF) learns the density of the clean data manifold. It identifies out-of-distribution embeddings, which are then corrected by a Geodesic Purification Solver. This solver projects adversarial points back onto the learned manifold via the shortest path, restoring a clean, semantically coherent representation. We conducted extensive evaluations on text classification (TC) across three datasets and multiple adversarial attacks. The results demonstrate that our method, MC^2F, not only establishes a new state-of-the-art in adversarial robustness but also fully preserves performance on clean data, even yielding modest gains in accuracy.
title Breaking the Adversarial Robustness-Performance Trade-off in Text Classification via Manifold Purification
topic Computation and Language
url https://arxiv.org/abs/2511.07888