Separating Representation from Reconstruction Enables Scalable Text Encoders
Abstract
Lay Summary
Today's AI language models come in two types. Decoders, which generate text and Encoders, which read a passage and turn it into a compact numerical summary of its meaning, powering search, recommendations, and the retrieval tools chatbots rely on. Encoders' core design has barely changed since the original BERT model appeared in 2018. When scaling up these models we found a surprising problem: as standard encoders grow, the summaries they produce become harder to use. Our solution, CrossBERT borrows an idea from computer vision to separate text understanding from text generation into two separate networks. One component focuses on understanding; a small separate one handles reconstruction. The result is higher-quality summaries, faster training, and steady improvement with scale.