Search papers, labs, and topics across Lattice.
The paper introduces CONE, a hybrid transformer encoder pre-trained model designed to encode numbers, ranges, and Gaussians into an embedding vector space while preserving distance. CONE uses a novel composite embedding construction algorithm that integrates numerical values, ranges, or Gaussians with their associated units and attribute names to capture their intricate semantics. Experiments across diverse datasets demonstrate CONE's strong numerical reasoning capabilities, achieving significant F1 score improvements over SOTA baselines, including a 9.37% improvement on the DROP dataset.
CONE's hybrid transformer architecture lets you encode numbers with units and attributes into embeddings that boost F1 scores on numerical reasoning tasks by up to 9.37% over existing models.
Large pre-trained models (LMs) and Large Language Models (LLMs) are typically effective at capturing language semantics and contextual relationships. However, these models encounter challenges in maintaining optimal performance on tasks involving numbers. Blindly treating numerical or structured data as terms is inadequate -- their semantics must be well understood and encoded by the models. In this paper, we propose CONE, a hybrid transformer encoder pre-trained model that encodes numbers, ranges, and gaussians into an embedding vector space preserving distance. We introduce a novel composite embedding construction algorithm that integrates numerical values, ranges or gaussians together with their associated units and attribute names to precisely capture their intricate semantics. We conduct extensive experimental evaluation on large-scale datasets across diverse domains (web, medical, finance, and government) that justifies CONE's strong numerical reasoning capabilities, achieving an F1 score of 87.28% on DROP, a remarkable improvement of up to 9.37% in F1 over state-of-the-art (SOTA) baselines, and outperforming major SOTA models with a significant Recall@10 gain of up to 25%.
