Stage-Aware Dual Alignment for Covariate Shift in Graph Domain Adaptation
Abstract
The performance of Graph Domain Adaptation ( GDA ) is fundamentally limited by covariate-driven discrepancies between source and target graphs, captured by Covariate Shift ( CS ) in the joint feature-structure space beyond label-related formulations. We show that CS induces bias at two non-interchangeable stages of Graph Neural Network ( GNN ) computation: Feature Shift ( FS ) distorts representations before message passing, while Feature-Conditional Structure Shift ( FCSS ) biases propagation, rendering single-stage alignment intrinsically insufficient. Based on this, we propose Dual Alignment for Covariate Shift ( DACS ), a stage-aware GDA framework that aligns discrepancies at their source. DACS mitigates FS via adversarial feature alignment and addresses FCSS through layer-wise reweighting to correct propagation bias, followed by final adversarial alignment for residual mismatch. This design follows directly from the stage-wise structure of CS rather than heuristic combinations of alignment modules. We further show that representation and propagation biases correspond to distinct components of target-domain error that cannot be eliminated in isolation. Experiments on synthetic and real-world benchmarks demonstrate that DACS consistently outperforms prior methods, especially under complex and coupled distribution shifts.
Related papers
See more related submissions when the market opens.