From 34422b5d3b5adea0d9ec08574d4c0ff4e25f1f9d Mon Sep 17 00:00:00 2001
From: Yoshitaka Inoue <8393063+inoue0426@users.noreply.github.com>
Date: Wed, 15 May 2024 22:09:04 -0400
Subject: [PATCH] Update README.md

---
 README.md | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index d5af982..e76842f 100644
--- a/README.md
+++ b/README.md
@@ -1,12 +1,23 @@
 # drGAT
 
+[![arXiv](https://img.shields.io/badge/arXiv-2405.08979-b31b1b.svg)](https://arxiv.org/abs/2405.08979)
+
 ![](Figs/Fig1.png)
 
 This is the official implementation for **drGAT: Attention-Guided Gene Assessment for Drug Response in Drug-Cell-Gene Heterogeneous Network**.  
 
 This model is created to understand how genes influence Drug Response using Graph Attention Networks (GAT) on heterogeneous networks of drugs, cells, and genes. It predicts Drug Response based on the attention coefficients generated during this process. This has been implemented in Python.
 
-
+```
+@misc{inoue2024drgat,
+      title={drGAT: Attention-Guided Gene Assessment of Drug Response Utilizing a Drug-Cell-Gene Heterogeneous Network}, 
+      author={Yoshitaka Inoue and Hunmin Lee and Tianfan Fu and Augustin Luna},
+      year={2024},
+      eprint={2405.08979},
+      archivePrefix={arXiv},
+      primaryClass={cs.LG}
+}
+```
 
 ## Requirement