Latent-optimized collaborative feature disentanglement for enhanced shadow removal
摘要
Shadows degrade image quality by obscuring textures, distorting colors, and reducing luminance. Existing methods often struggle to fully restore shadowed regions due to their reliance on basic illumination enhancement. This paper introduces VAE-mSRNet, a Collaborative Feature Disentanglement network that extends the multi-scale shadow removal framework mSRNet and decomposes shadowed and shadow-free images into shared and distinct feature components. By integrating a Variational Autoencoder (VAE), VAE-mSRNet compactifies the feature space and strengthens the separation of illumination-invariant textures and color cues. This results in a more structured representation that improves both separation quality and feature reconstruction. Our approach addresses challenges in shadow removal, such as texture inconsistency and color imbalance. Comprehensive experiments on benchmark datasets (AISTD and SRD) demonstrate that VAE-mSRNet achieves competitive and in many cases leading results across PSNR and RMSE metrics, with particular strength in shadow-region restoration, while maintaining fine texture details and color fidelity in complex shadow scenarios. Here, we show that VAE-mSRNet offers a robust solution for shadow removal, setting a new benchmark in the field. The source code and pre-trained models are available at https://github.com/Midrees330/VAE_MSRNet.