Instructions to use aehrc/cxrmate-rrg24 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use aehrc/cxrmate-rrg24 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="aehrc/cxrmate-rrg24", trust_remote_code=True)# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("aehrc/cxrmate-rrg24", trust_remote_code=True) model = AutoModel.from_pretrained("aehrc/cxrmate-rrg24", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -8,10 +8,10 @@ datasets:
|
|
| 8 |
|
| 9 |
# CXRMate-RRG4: Entropy-Augmented Self-Critical Sequence Training for Radiology Report Generation
|
| 10 |
|
| 11 |
-
This is an evolution of https://huggingface.co/aehrc/cxrmate developed for the Radiology Report Generation task of BioNLP @ ACL 2024.
|
| 12 |
|
| 13 |
For this, we proposed EAST: Entropy-Augmented Self-critical sequence Training (EAST):
|
| 14 |
-
- EAST modifies Self-Critical Sequence Training (SCST) by adding entropy regularisation.
|
| 15 |
- Helps maintain a higher entropy in the token distribution.
|
| 16 |
- Preventing overfitting to common phrases and ensuring a broader exploration of the vocabulary during training.
|
| 17 |
- This was essential to handle the diversity of the radiology reports in the RRG24 datasets.
|
|
|
|
| 8 |
|
| 9 |
# CXRMate-RRG4: Entropy-Augmented Self-Critical Sequence Training for Radiology Report Generation
|
| 10 |
|
| 11 |
+
This is an evolution of https://huggingface.co/aehrc/cxrmate developed for the [Radiology Report Generation](https://stanford-aimi.github.io/RRG24/) task of [BioNLP @ ACL 2024](https://aclweb.org/aclwiki/BioNLP_Workshop).
|
| 12 |
|
| 13 |
For this, we proposed EAST: Entropy-Augmented Self-critical sequence Training (EAST):
|
| 14 |
+
- EAST modifies [Self-Critical Sequence Training (SCST)](https://openaccess.thecvf.com/content_cvpr_2017/papers/Rennie_Self-Critical_Sequence_Training_CVPR_2017_paper.pdf) by adding entropy regularisation.
|
| 15 |
- Helps maintain a higher entropy in the token distribution.
|
| 16 |
- Preventing overfitting to common phrases and ensuring a broader exploration of the vocabulary during training.
|
| 17 |
- This was essential to handle the diversity of the radiology reports in the RRG24 datasets.
|