Saved in:
Bibliographic Details
Main Authors: Liu, Hong, Lu, Yitong
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2411.16236
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909403744043008
author Liu, Hong
Lu, Yitong
author_facet Liu, Hong
Lu, Yitong
contents This paper presents a novel method to improve the robustness of foundation models to group-based biases. We propose a simple yet effective method, called DoubleCCA, that leverages random sentences and Canonical Correlation Analysis (CCA) to enrich the text embeddings of the foundation model. First, we generate various random sentences that augment the original prompts, which extends the original prompts with random words or character sequences. Second, we use an additional sentence embedding model to generate different text embeddings with respect to these random sentences. We then use CCA double twice to align the representations and reconstruct them back to the original representation space. We demonstrate the effectiveness of our method on a variety of tasks and datasets, showing that it outperforms existing methods in terms of both performance and robustness. Our method is simple to implement and can be easily integrated into existing models, making it a practical solution for improving the robustness of foundation models to group-based biases.
format Preprint
id arxiv_https___arxiv_org_abs_2411_16236
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle DoubleCCA: Improving Foundation Model Group Robustness with Random Sentence Embeddings
Liu, Hong
Lu, Yitong
Computation and Language
Computer Vision and Pattern Recognition
This paper presents a novel method to improve the robustness of foundation models to group-based biases. We propose a simple yet effective method, called DoubleCCA, that leverages random sentences and Canonical Correlation Analysis (CCA) to enrich the text embeddings of the foundation model. First, we generate various random sentences that augment the original prompts, which extends the original prompts with random words or character sequences. Second, we use an additional sentence embedding model to generate different text embeddings with respect to these random sentences. We then use CCA double twice to align the representations and reconstruct them back to the original representation space. We demonstrate the effectiveness of our method on a variety of tasks and datasets, showing that it outperforms existing methods in terms of both performance and robustness. Our method is simple to implement and can be easily integrated into existing models, making it a practical solution for improving the robustness of foundation models to group-based biases.
title DoubleCCA: Improving Foundation Model Group Robustness with Random Sentence Embeddings
topic Computation and Language
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2411.16236