A JEPA-Inspired Span-Masked Framework for Language Representation Learning: Revisiting Cosine Similarity and VICReg Regularization

Joint-Embedding Predictive Architectures (JEPAs) have recently emerged as a promising paradigm for self-supervised representation learning by predicting latent embeddings from partial observations rather than reconstructing raw inputs. Although JEPA has demonstrated considerable success in computer vision and large-scale language models, the design of lightweight JEPA-inspired representation learning frameworks for natural language remains insufficiently understood. This paper proposes a lightweight span-masked JEPA-inspired framework that formulates language representation learning as latent semantic prediction from partially observed contextual inputs. The framework predicts target embeddings generated by a momentum-updated encoder while avoiding direct token reconstruction, thereby emphasizing semantic abstraction over lexical recovery. To investigate the influence of objective function design, cosine similarity and Variance–Invariance–Covariance Regularization (VICReg) are systematically compared under identical architectural and training conditions, with and without auxiliary masked language modeling (MLM) supervision. The proposed framework is evaluated on one controlled dataset and six benchmark datasets, including AG News, SST-2, IMDb, TREC, DBPedia, and Yelp Polarity. Experimental results consistently demonstrate that the cosine-based JEPA objective without auxiliary MLM supervision provides the best balance between representation quality, computational efficiency, and downstream linear probing performance. Statistical significance analysis using the Friedman test and pairwise Wilcoxon signed-rank tests with Holm correction further confirms that the cosine-based objective significantly outperforms the remaining configurations, while no significant difference is observed between VICReg with and without MLM supervision. The findings indicate that latent semantic prediction alone is sufficient for learning robust language representations under the proposed lightweight framework, whereas auxiliary token-level reconstruction does not consistently improve downstream performance despite increasing computational cost. Overall, this work provides reproducible empirical evidence and practical guidance for objective function design in lightweight JEPA-inspired language representation learning, offering an efficient experimental foundation for future self-supervised language models.

Authors

Institutions

Publication Details

Journal
Applied Sciences
Published
2026-08-27
DOI
https://doi.org/10.3390/app16178530
Primary Topic
Topic Modeling
Type
article
Field-Weighted Citation Impact
0.00

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

A JEPA-Inspired Span-Masked Framework for Language Representation Learning: Revisiting Cosine Similarity and VICReg Regularization

Khanabhorn Kawattikul, Chatklaw Jareanpon
Applied Sciences
Topic Modeling
article

A JEPA-Inspired Span-Masked Framework for Language Representation Learning: Revisiting Cosine Similarity and VICReg Regularization

Khanabhorn Kawattikul, Chatklaw Jareanpon
article en

Abstract

Joint-Embedding Predictive Architectures (JEPAs) have recently emerged as a promising paradigm for self-supervised representation learning by predicting latent embeddings from partial observations rather than reconstructing raw inputs. Although JEPA has demonstrated considerable success in computer vision and large-scale language models, the design of lightweight JEPA-inspired representation learning frameworks for natural language remains insufficiently understood. This paper proposes a lightweight span-masked JEPA-inspired framework that formulates language representation learning as latent semantic prediction from partially observed contextual inputs. The framework predicts target embeddings generated by a momentum-updated encoder while avoiding direct token reconstruction, thereby emphasizing semantic abstraction over lexical recovery. To investigate the influence of objective function design, cosine similarity and Variance–Invariance–Covariance Regularization (VICReg) are systematically compared under identical architectural and training conditions, with and without auxiliary masked language modeling (MLM) supervision. The proposed framework is evaluated on one controlled dataset and six benchmark datasets, including AG News, SST-2, IMDb, TREC, DBPedia, and Yelp Polarity. Experimental results consistently demonstrate that the cosine-based JEPA objective without auxiliary MLM supervision provides the best balance between representation quality, computational efficiency, and downstream linear probing performance. Statistical significance analysis using the Friedman test and pairwise Wilcoxon signed-rank tests with Holm correction further confirms that the cosine-based objective significantly outperforms the remaining configurations, while no significant difference is observed between VICReg with and without MLM supervision. The findings indicate that latent semantic prediction alone is sufficient for learning robust language representations under the proposed lightweight framework, whereas auxiliary token-level reconstruction does not consistently improve downstream performance despite increasing computational cost. Overall, this work provides reproducible empirical evidence and practical guidance for objective function design in lightweight JEPA-inspired language representation learning, offering an efficient experimental foundation for future self-supervised language models.

Applied SciencesVol. 16(17)
Mahasarakham University (TH), Rajamangala University of Technology Isan (TH), Rajamangala University of Technology Tawan-ok (TH), Rambhai Barni Rajabhat University (TH)
Mahasarakham University
Quality Education
Openalex Percentile: Top 8%
Topic Modeling
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.