REZE: Representation Regularization for Domain-adaptive Text Embedding Pre-finetuning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lee, Seungmin, Lee, Jeonghwan, Lim, Hyunkuk, Kim, Sejoon, Sung, Mingi
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913050559250432
author Lee, Seungmin
Lee, Jeonghwan
Lim, Hyunkuk
Kim, Sejoon
Sung, Mingi
author_facet Lee, Seungmin
Lee, Jeonghwan
Lim, Hyunkuk
Kim, Sejoon
Sung, Mingi
contents Recent text embedding models are often adapted to specialized domains via contrastive pre-finetuning (PFT) on a naive collection of scattered, heterogeneous tasks. However, this approach often introduces task-induced bias alongside domain knowledge, leading to uncontrolled representation shifts that distort the pretrained embedding geometry and cause substantial performance degradation. To address this issue, we propose REZE, a representation regularization framework that explicitly controls representation shift during embedding pre-finetuning. REZE operates on the relations of anchor-positive pairs and decomposes them in an eigenspace. It then measures task-wise dispersion along each eigencomponent to identify task-variant directions and applies adaptive soft-shrinkage to suppress task-induced noise while preserving task-invariant semantic structure, without inference-time overhead. Experiments across multiple embedding backbones and specialized benchmarks show that REZE outperforms standard pre-finetuning and isotropy-oriented post-hoc regularization in most settings, remaining stable where existing PFT variants collapse. Embedding space analyses further confirm that REZE induces controlled shifts aligned with the original embedding manifold, underscoring representation shift control as a key principle for robust embedding pre-finetuning under heterogeneous supervision.
format Preprint
id arxiv_https___arxiv_org_abs_2604_17257
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle REZE: Representation Regularization for Domain-adaptive Text Embedding Pre-finetuning
Lee, Seungmin
Lee, Jeonghwan
Lim, Hyunkuk
Kim, Sejoon
Sung, Mingi
Computation and Language
Artificial Intelligence
Recent text embedding models are often adapted to specialized domains via contrastive pre-finetuning (PFT) on a naive collection of scattered, heterogeneous tasks. However, this approach often introduces task-induced bias alongside domain knowledge, leading to uncontrolled representation shifts that distort the pretrained embedding geometry and cause substantial performance degradation. To address this issue, we propose REZE, a representation regularization framework that explicitly controls representation shift during embedding pre-finetuning. REZE operates on the relations of anchor-positive pairs and decomposes them in an eigenspace. It then measures task-wise dispersion along each eigencomponent to identify task-variant directions and applies adaptive soft-shrinkage to suppress task-induced noise while preserving task-invariant semantic structure, without inference-time overhead. Experiments across multiple embedding backbones and specialized benchmarks show that REZE outperforms standard pre-finetuning and isotropy-oriented post-hoc regularization in most settings, remaining stable where existing PFT variants collapse. Embedding space analyses further confirm that REZE induces controlled shifts aligned with the original embedding manifold, underscoring representation shift control as a key principle for robust embedding pre-finetuning under heterogeneous supervision.
title REZE: Representation Regularization for Domain-adaptive Text Embedding Pre-finetuning
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2604.17257