From c1479791ca4376f47d98b008bab0a2e9fe800055 Mon Sep 17 00:00:00 2001
From: Mackenzie Mathis <mackenzie.mathis@epfl.ch>
Date: Sun, 3 Nov 2024 00:16:37 +0100
Subject: [PATCH] Update README.md

---
 README.md | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/README.md b/README.md
index 17ffa42..458a1cf 100644
--- a/README.md
+++ b/README.md
@@ -1,5 +1,9 @@
 # AROS: Adversarially Robust Out-of-Distribution Detection through Stability
 
+
+[![PyPI version](https://badge.fury.io/py/aros-node.svg)](https://badge.fury.io/py/aros-node)
+![License: Apache 2.0](https://img.shields.io/badge/License-Apache_2.0-purple)
+
 ## Overview
 
 This repository contains the code for the paper **"Adversarially Robust Out-of-Distribution Detection Using Lyapunov-Stabilized Embeddings"**. The method, termed **AROS**, employs Neural Ordinary Differential Equations (NODEs) with Lyapunov stability to create robust embeddings for OOD detection, significantly improving performance against adversarial attacks.
@@ -7,6 +11,10 @@ This repository contains the code for the paper **"Adversarially Robust Out-of-D
 
 ![AROS](https://github.com/user-attachments/assets/b0d9e7f8-e39d-4bae-aee2-79a247b5e87f)
 
+## Preprint
+
+Check out our preprint now out on arXiv! [Adversarially Robust Out-of-Distribution Detection Using Lyapunov-Stabilized Embeddings](https://arxiv.org/abs/2410.10744). Hossein Mirzaeri & Mackenzie W. Mathis. Oct 2024.
+
  
 ## Key Features
 
@@ -37,6 +45,28 @@ This repository contains the code for the paper **"Adversarially Robust Out-of-D
 ## Installation
 
 To install the necessary packages, run:
+```python
+pip install git+https://github.com/RobustBench/robustbench.git
+pip install aros-node
+```
+
+To install the necessary packages from source (locally), run:
 
 ```bash
 pip install -r requirements.txt
+```
+
+## Citation
+```
+@article{mirzaei2024aros,
+      title={Adversarially Robust Out-of-Distribution Detection Using Lyapunov-Stabilized Embeddings}, 
+      author={Hossein Mirzaei and Mackenzie W. Mathis},
+      year={2024},
+      eprint={2410.10744},
+      journal={arXiv},
+      primaryClass={cs.LG},
+      url={https://arxiv.org/abs/2410.10744}, 
+}
+```
+
+