Search papers, labs, and topics across Lattice.
The paper introduces MeInTime, a diffusion-based face restoration method that extends reference-based techniques to handle cross-age scenarios, addressing the limitation of existing methods that assume age-aligned inputs. MeInTime decouples identity and age modeling, using a novel attention mechanism and Gated Residual Fusion for identity feature injection during training. At inference, Age-Aware Gradient Guidance is employed to steer the denoising process towards the desired age, resulting in improved identity preservation and age consistency compared to existing methods.
Restoring faces across age gaps is now possible: MeInTime leverages diffusion models and age-aware guidance to create faithful restorations from cross-age references.
To better preserve an individual's identity, face restoration has evolved from reference-free to reference-based approaches, which leverage high-quality reference images of the same identity to enhance identity fidelity in the restored outputs. However, most existing methods implicitly assume that the reference and degraded input are age-aligned, limiting their effectiveness in real-world scenarios where only cross-age references are available, such as historical photo restoration. This paper proposes MeInTime, a diffusion-based face restoration method that extends reference-based restoration from same-age to cross-age settings. Given one or few reference images along with an age prompt corresponding to the degraded input, MeInTime achieves faithful restoration with both identity fidelity and age consistency. Specifically, we decouple the modeling of identity and age conditions. During training, we focus solely on effectively injecting identity features through a newly introduced attention mechanism and introduce Gated Residual Fusion modules to facilitate the integration between degraded features and identity representations. At inference, we propose Age-Aware Gradient Guidance, a training-free sampling strategy, using an age-driven direction to iteratively nudge the identity-aware denoising latent toward the desired age semantic manifold. Extensive experiments demonstrate that MeInTime outperforms existing face restoration methods in both identity preservation and age consistency. Our code is available at: https://github.com/teer4/MeInTime
