atlas

Research map / Architectures and efficiency / Attention and transformers

BiPFT: Binary Pre-trained Foundation Transformer with Low-Rank Estimation of Binarization Residual Polynomials

AAAI 2024 · Architectures and efficiency › Attention and transformers · approach: Transformer neural operators for PDEs · 3 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes BiPFT, the first binary pretrained foundation transformer for NLU, pretrained with 1-bit weights and activations plus a low-rank data-driven estimator of self-attention binarization residual polynomials, and finetuned without task-specific distillation.

Problem

Task-specific binary BERTs rely on full-precision teacher distillation and heavy hyperparameter search, are unstable, and lose learning ability without distillation; there is no binary foundation model for NLU.

Approach

Pretrain a binary BERT-style transformer (BiPFT-A) with MLM, NSP and task-agnostic distillation so finetuning needs no FP teacher; then analyze binarization error in self-attention as residual polynomials and model them with rank-1 low-rank estimators (BiPFT-B) learned during pretraining.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
binary pretraining rather than training the binary baseline from scratchBinary BERT · GLUE devaverage score: 55.4 -> 69.2Helped
add low-rank residual polynomial estimators to BiPFT-ABiPFT-A · GLUE devaverage score: 69.2 -> 70.8Helped
add attention-value estimator only (ablation)BiPFT-A · GLUE devaverage score: 69.2 -> 69.9Helped
add key-query estimator only (ablation)BiPFT-A · GLUE devaverage score: 69.2 -> 70.2Helped
combine key-query and attention-value estimators (ablation)BiPFT-A · GLUE devaverage score: 69.2 -> 70.8Helped
add estimators without binary pretraining (ablation)Baseline · GLUE devaverage score: 55.4 -> 54.9Hurt
increase estimator rank from 1 to 2 (ablation)BiPFT-B · GLUE devaverage score: 70.8 -> 69.7Hurt
increase estimator rank from 1 to 4 (ablation)BiPFT-B · GLUE devaverage score: 70.8 -> 69.2Hurt
add LoRA alone (ablation)BiPFT-A · GLUE devaverage score: 69.2 -> 69.9Helped
combine LoRA with residual polynomial estimators (ablation)BiPFT-B · GLUE devaverage score: 70.8 -> 68.0Hurt
use full-precision pretrained initialization (ablation)BiPFT-B · GLUE devaverage score: 69.6 -> 70.8Helped
use full-precision task-specific initialization (ablation)Baseline* · GLUE devaverage score: 61.6 -> 68.1Helped

Surprising findings

Limitations

Experimental setup

Datasets: BooksCorpus English Wikipedia GLUE

Benchmarks: GLUE

Metrics: accuracy F1 Pearson/Spearman correlation average GLUE score FLOPs model size

Baselines: BiT BiBERT BinaryBERT TernaryBERT Q-BERT Q2BERT baseline*

Builds on: BiT BiBERT BinaryBERT BERT LoRA

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy