MARFI: Missingness-Aware Rectified-Flow Imputation for MNAR Data in Predictive Maintenance
Results indicate that explicitly using the missingness mask in the restoration process can improve numerical imputation performance while also preserving information useful for downstream prediction under missing-data settings relevant to predictive maintenance.