Paper Detail

PIcsC: Partitioning-Induced Covariate Shift Correction

Behraj Khan, Behroz Mirza, Syed Ahmad Chan Bukhari, Tahir Qasim Syed

arxiv Score 9.0

Published 2026-07-28 · First seen 2026-07-31

Research Track A · General AI

Abstract

Covariate shift across training-data partitions biases model selection and parameter estimation in cross-validation, lifelong learning, and federated learning. We propose \textit{Partition-Induced Covariate-shift Correction} (\texttt{PIcsC}), a Fisher information-based regularization framework that mitigates distribution mismatch between data partitions and a reference distribution. \texttt{PIcsC} approximates partition divergence using the Fisher Information Matrix (FIM) and incorporates the resulting statistic as a regularizer during optimization. The same formulation applies to both centrally partitioned datasets (batches or cross-validation folds) and inherently distributed data (federated clients or decentralized nodes), requiring only partition-local gradient statistics rather than raw data. We further introduce a conditional adaptation mechanism that combines FIM shift with KL divergence to detect significant distribution shifts and activates regularization only when necessary. Experiments on more than 40 datasets demonstrate consistent improvements under both natural and synthetic covariate shift. On fragmented batch and fold settings, \texttt{PIcsC} reduces fragmentation-induced performance degradation by more than 20\% and 25\%, respectively. On seven federated learning benchmarks, it consistently outperforms FedAvg, FedProx, and SCAFFOLD by 3 -5 percentage points without requiring client-specific personalization. These results demonstrate that Fisher information provides an effective and unified mechanism for mitigating partition-induced covariate shift across both centralized and distributed learning.

Workflow Status

Review status
pending
Role
unreviewed
Read priority
soon
Vote
Not set.
Saved
no
Collections
Not filed yet.
Next action
Not filled yet.

Reading Brief

No structured notes yet. Add `summary_sections`, `why_relevant`, `claim_impact`, or `next_action` in `papers.jsonl` to enrich this view.

Why It Surfaced

No ranking explanation is available yet.

Tags

No tags.

BibTeX

@article{khan2026picsc,
  title = {PIcsC: Partitioning-Induced Covariate Shift Correction},
  author = {Behraj Khan and Behroz Mirza and Syed Ahmad Chan Bukhari and Tahir Qasim Syed},
  year = {2026},
  abstract = {Covariate shift across training-data partitions biases model selection and parameter estimation in cross-validation, lifelong learning, and federated learning. We propose \textbackslash{}textit\{Partition-Induced Covariate-shift Correction\} (\textbackslash{}texttt\{PIcsC\}), a Fisher information-based regularization framework that mitigates distribution mismatch between data partitions and a reference distribution. \textbackslash{}texttt\{PIcsC\} approximates partition divergence using the Fisher Information Matrix (FIM) and incorporates the re},
  url = {https://arxiv.org/abs/2607.25441},
  keywords = {cs.LG},
  eprint = {2607.25441},
  archiveprefix = {arXiv},
}

Metadata

{}