CDAE: Enhancing Perturbation Robustness in Pretrained Language Models with Contrastive Denoising
Abstract
Pre-trained language models have significantly improved sentence representation learning, yet their embedding remain sensitive to semantic preserving textual perturbations such as synonym substitution, masking and word dropout. This work proposes a lightweight Contrastive Denoising Autoencoder (CDAE) that refines pre-trained BERT embedding by jointly optimizing contrastive and reconstruction objective to learn perturbation-invariant representation. We evaluate the proposed framework using multiple perturbation strategies with varying strengths and compare it against the original BERT embeddings and SimCSE. Experimental results show that CDAE consistently preserves higher embedding similarity under perturbations, with the improvements becoming more pronounced as framework effectively enhances representation stability while preserving semantic information, highlighting perturbation-invariant learning as a promising direction for improving sentence embeddings. The source code is publicly available at: https://github.com/ComputationIASBS/CDAE
Cite
@article{arxiv.2607.28236,
title = {CDAE: Enhancing Perturbation Robustness in Pretrained Language Models with Contrastive Denoising},
author = {Sina Heydari and Amirreza Abbasi and Mohsen Hooshmand and Majid Ramezani},
journal= {arXiv preprint arXiv:2607.28236},
year = {2026}
}
Comments
Submitted to 16th International Conference on Computer and Knowledge Engineering (ICCKE 2026)