English

PI-NAIM: Path-Integrated Neural Adaptive Imputation Model

Computer Vision and Pattern Recognition 2025-11-18 v1 Artificial Intelligence

Abstract

Medical imaging and multi-modal clinical settings often face the challange of missing modality in their diagnostic pipelines. Existing imputation methods either lack representational capacity or are computationally expensive. We propose PI-NAIM, a novel dual-path architecture that dynamically routes samples to optimized imputation approaches based on missingness complexity. Our framework integrates: (1) intelligent path routing that directs low missingness samples to efficient statistical imputation (MICE) and complex patterns to powerful neural networks (GAIN with temporal analysis); (2) cross-path attention fusion that leverages missingness-aware embeddings to intelligently combine both branches; and (3) end-to-end joint optimization of imputation accuracy and downstream task performance. Extensive experiments on MIMIC-III and multimodal benchmarks demonstrate state-of-the-art performance, achieving RMSE of 0.108 (vs. baselines' 0.119-0.152) and substantial gains in downstream tasks with an AUROC of 0.812 for mortality prediction. PI-NAIM's modular design enables seamless integration into vision pipelines handling incomplete sensor measurements, missing modalities, or corrupted inputs, providing a unified solution for real-world scenario. The code is publicly available at https://github.com/AfifaKhaled/PI-NAIM-Path-Integrated-Neural-Adaptive-Imputation-Model

Keywords

Cite

@article{arxiv.2511.11908,
  title  = {PI-NAIM: Path-Integrated Neural Adaptive Imputation Model},
  author = {Afifa Khaled and Ebrahim Hamid Sumiea},
  journal= {arXiv preprint arXiv:2511.11908},
  year   = {2025}
}
R2 v1 2026-07-01T07:38:30.525Z