Update README.md

README.md

@@ -11,6 +11,9 @@ pipeline_tag: feature-extraction
 
 IDs-C2I-Dec is a bi-encoder retrieval model specifically fine-tuned for the IDs subset of the CodeConvo dataset. (IDs is short for Internet-Drafts)
 
+- **Paper:** [Automated Insights Into GitHub Collaboration Dynamics](https://ieeexplore.ieee.org/stamp/stamp.jsp?tp=&arnumber=10982058) 
+- **Training Dataset:** [jiebi/CodeConvo](https://huggingface.co/datasets/jiebi/CodeConvo)
+
 ## Uses
 
 <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
@@ -21,11 +24,21 @@ To run the IR evaluation task ([source code](https://github.com/cheop-byeon/mteb
 
 ## Training Details
 
-### Training Data
-
-<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
-IDs Subset of CodeConvo https://huggingface.co/datasets/jiebi/CodeConvo
+You can reproduce this with [this script](https://github.com/cheop-byeon/FlagEmbedding/blob/main/examples/finetune/embedder/decoder_only/ft_CodeConvo_decoder.sh) (reproduced here for convenience).
 
 ### Framework versions
 
-- PEFT 0.13.2
+- PEFT 0.13.2
+
+### Citation
+
+```bibtex
+@article{bian2025automated,
+  title={Automated Insights Into GitHub Collaboration Dynamics},
+  author={Bian, Jie and Arefev, Nikolay and M{\"u}hlh{\"a}user, Max and Welzl, Michael},
+  journal={IEEE Access},
+  volume={13},
+  pages={85526--85542},
+  year={2025}
+}
+```