Search papers, labs, and topics across Lattice.
This paper introduces Confusion Geometry Rebalancing (CGRm), a novel framework for improving adversarial training in long-tailed distributions by addressing dual imbalances in class representation and adversarial maximization. By utilizing directed robust errors and a confusion geometry graph, CGRm optimizes class-wise loss weights and enhances the robustness of underrepresented classes. Experimental results demonstrate that CGRm significantly outperforms existing methods, providing consistent performance gains on long-tailed benchmarks while validating the effectiveness of its components through ablation studies.
Long-tailed adversarial training can be revolutionized by leveraging confusion geometry to boost the robustness of vulnerable classes and sharpen critical decision boundaries.
Adversarial training under long tailed distributions suffers from a dual imbalance: the class imbalance skews the training objective toward head classes, and the adversarial inner maximization may further amplify this bias. Existing methods mitigate this issue by correcting class priors or adapting class wise robust supervision, yet they treat each class in isolation and fail to identify which boundaries drive long tailed collapse. We propose a Confusion Geometry Rebalancing method (CGRm) for long tail adversarial training, a plug in framework that leverages directed robust errors as training signals. CGRm leverages periodic robust evaluations to derive source class loss weights, class wise robust coefficients, and a directed confusion geometry graph. The method then couples feedback weighted robust optimization with graph guided margin correction, thereby boosting the robustness of vulnerable classes and sharpening the critical boundaries that drive long tailed performance degradation. Experiments on long tailed benchmarks show that CGRm achieves consistent robust performance gains over existing methods, with ablations validating the contribution of each component. We provide the code in the supplement.
