atlas

Research map / Label-efficient and robust learning / Recognition and long-tail

RMIB: Representation Matching Information Bottleneck for Matching Text Representations

ICML 2024 · Label-efficient and robust learning › Recognition and long-tail · approach: Multilingual language representation pretraining

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 RMIB, an information-bottleneck extension for asymmetric-domain text matching that aligns text representations to a Gaussian prior and maximizes interaction between the two representations, improving four matching models on five datasets.

Problem

Asymmetric-domain text matching: how to align text representation distributions across domains and what role this alignment plays; existing IB does not constrain interaction between the two text representations.

Approach

Narrow each text representation's distribution by matching it to a shared standard Gaussian prior, showing this equals optimizing the information bottleneck; extend IB with interaction and inadequacy constraints so the objective becomes maximizing conditional mutual information I(Z1;Z2|Y), which is approximated via a MINE-style bound.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add RMIB to BERTBERT · SICKaccuracy: 86.69 -> 87.40Helped
Add RMIB to BERTBERT · SciTailaccuracy: 93.79 -> 94.12Helped
Add RMIB to BERTBERT · WikiQAF1: 94.88 -> 95.16Helped
Add RMIB to BERTBERT · WikiQAMAP: 83.40 -> 84.29Helped
Add RMIB to BERTBERT · WikiQAMRR: 85.01 -> 86.02Helped
Add RMIB to BERTBERT · SNLIaccuracy: 90.46 -> 90.43Hurt
Add RMIB to SBERTSBERT · Quora Question Pairaccuracy: 86.46 -> 87.34Helped
Add RMIB to BERTBERT · MRPCaccuracy: 85.54 -> 87.01Helped
Add RMIB to BERTBERT · WNLIaccuracy: 53.52 -> 56.34Helped
Add RMIB to BERTBERT · RTEaccuracy: 62.82 -> 69.31Helped
Remove the prior-distribution KL term (set α1 to 0) (ablation)RMIB · multiple text matching datasetsaccuracyMixed
Remove the representation-interaction term (set α2 to 0) (ablation)RMIB · multiple text matching datasetsaccuracyMixed

Surprising findings

Limitations

Experimental setup

Datasets: SICK SciTail WikiQA SNLI Quora Question Pairs MRPC RTE WNLI

Benchmarks: SICK SciTail WikiQA SNLI Quora Question Pairs MRPC RTE WNLI

Metrics: accuracy F1 MAP MRR

Baselines: RE2 BERT SBERT ESIM DDR-Match (JS) DDR-Match (MMD) DDR-Match (WD) IB (RMIB ablation)

Builds on: information bottleneck (Tishby et al., 2000) VIB DDR-Match MINE RE2 ESIM

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