Skip to content

Repository files navigation

Code of CGE

introduction

A code for What's New in My Data? Novelty Exploration via Contrastive Generation @ ICLR 2025

Authors:
Masaru Isonuma1,2,3 and Ivan Titov1,4
1University of Edinburgh 2University of Tokyo 3National Institute of Informatics
4University of Amsterdam

Environment

Python==3.8

  • Run the following command to install the required packages:
pip install -r requirements.txt
  • Or you can reproduce the environment if you use conda:
conda env create -n cge -f cge.yml
conda activate untrac

Discovering the novel domains in the fine-tuning dataset via CGE

Run CGE with the following commands. The generated texts are displayed at each iteration in the console. You can try the static version of CGE by setting --no_update in the arguments of main.py.

  • Run CGE for Falcon-RW fine-tuned on the code dataset.
bash run_falcon.sh
  • Run CGE for OpenLlama fine-tuned on the non-English dataset.
bash run_llama.sh

About

https://arxiv.org/abs/2410.14765

Resources

Stars

1 star

Watchers

1 watching

Forks

Releases

Packages

Used by

Contributors

Languages