cs.CVAug 10, 2026

C2^2A: Coupling Spatial Evidence with Clinical Priors via Co-occurrence Aware Class Attention for Multi-Label Chest X-Ray Classification

Authors: Akash GogineniNagur Shareef ShaikAasrith MandavaAdnan MasoodDong Hye Ye

Organizations: Department of Computer Science, Georgia State University, Atlanta, GA 30324, USA · UST Global Inc., Aliso Viejo, CA 92656, USA

Abstract

Thoracic pathologies rarely occur in isolation, yet standard multi-label classifiers rely on shared global descriptors, discarding \emph{where} findings lie and \emph{how} they co-occur. We propose \textbf{C2\mathbf{^2}A} (Co-occurrence Aware Class Attention), a classification head that explicitly couples spatial evidence with clinical priors. First, C2^2A casts pooling as an expectation over learned per-class spatial attention maps, yielding localized descriptors for each disease. Second, it couples these descriptors via a learnable graph warm-started from empirical label co-occurrence. A single residual message-passing step shares evidence among related findings, proving to be a bounded perturbation of the identity where co-occurrence enters each logit through an explicit bilinear interaction. On CheXpert, C2^2A achieves a superior 0.8950.895 macro-mean AUROC, outperforming advanced context-gating baselines. Crucially, gains concentrate on highly co-occurrent classes with ambiguous spatial evidence (rescuing Atelectasis by +1.5+1.5 over GCG), demonstrating the prior's regularizing effect with a negligible overhead of one linear projection and a C ⁣× ⁣CC\!\times\!C edge matrix.

Explore similar work

CardsList