Reversible data hiding in encrypted images based on pixel-level masked autoencoder and polar code

Abstract

In the study of vacating-room-after-encryption reversible data hiding in encrypted images (VRAE RDHEI), pixel prediction is an important mechanism to achieve reversibility, which has a crucial impact on the capacity and fidelity. In this paper, we propose a novel pixel-level masked autoencoders (PLMAE) as a high-performance pixel predictor for RDHEI. Unlike the original masked autoencoders (MAE), PLMAE focuses on pixel-level reconstruction rather than semantic patch-level reconstruction. The purpose of PLMAE is to spare more carrier pixels while maintaining relatively high prediction accuracy, thereby improving the RDHEI capacity. Based on PLMAE, a novel RDHEI method is proposed. In the proposed method, the data hider encodes the secret data using a polar code and then embeds the encoded data. After the image is decrypted, the receiver considers the carrier pixels as masked pixels, predicts the original states of the carrier pixels using PLMAE to extract the secret data, and then decodes the secret data and recovers the image based on the decoding results. The experimental results demonstrate that the proposed method in this paper can achieve better performance than the existing methods.