Improve model card: Add metadata, paper/project/code links, and correct dataset usage

README.md

@@ -1,6 +1,21 @@
+---
+license: apache-2.0
+pipeline_tag: image-classification
+library_name: pytorch
+tags:
+  - fashionmnist
+  - resnet
+  - unlearning
+---
+
 # Model Card for Yurim0507/resnet18-fashionmnist-unlearning
 
-This repository contains ResNet-18 models retrained on the FashionMNIST dataset with specific classes excluded during training. Each model is trained to study the impact of excluding a class on model performance and generalization on the remaining classes.
+This repository contains ResNet-18 models retrained on the FashionMNIST dataset with specific classes excluded during training, as part of the research presented in the paper [Unlearning Comparator: A Visual Analytics System for Comparative Evaluation of Machine Unlearning Methods](https://huggingface.co/papers/2508.12730).
+
+**📚 Paper** | **🌐 Project Page** | **💻 GitHub Repository**
+[https://huggingface.co/papers/2508.12730](https://huggingface.co/papers/2508.12730) | [https://gnueaj.github.io/Machine-Unlearning-Comparator/](https://gnueaj.github.io/Machine-Unlearning-Comparator/) | [https://github.com/gnueaj/Machine-Unlearning-Comparator](https://github.com/gnueaj/Machine-Unlearning-Comparator)
+
+Each model is trained to study the impact of excluding a class on model performance and generalization on the remaining classes.
 
 ## Evaluation
 - **Testing Data**: FashionMNIST test set (10,000 images, 1,000 per class)  
@@ -11,17 +26,17 @@ This repository contains ResNet-18 models retrained on the FashionMNIST dataset
 
 ### Results
 | Model File                               | Excluded Class   | FashionMNIST Accuracy |
-|------------------------------------------|------------------|-----------------------|
-| `resnet18_fashionmnist_original.pth`     | None             | 95.39%                |
-| `resnet18_fashionmnist_forget0.pth`      | T-shirt/top      | 96.54%                |
-| `resnet18_fashionmnist_forget1.pth`      | Trouser          | 95.07%                |
-| `resnet18_fashionmnist_forget2.pth`      | Pullover         | 95.99%                |
-| `resnet18_fashionmnist_forget3.pth`      | Dress            | 95.69%                |
-| `resnet18_fashionmnist_forget4.pth`      | Coat             | 95.92%                |
-| `resnet18_fashionmnist_forget5.pth`      | Sandal           | 94.88%                |
-| `resnet18_fashionmnist_forget6.pth`      | Shirt            | 97.76%                |
-| `resnet18_fashionmnist_forget7.pth`      | Sneaker          | 95.37%                |
-| `resnet18_fashionmnist_forget8.pth`      | Bag              | 94.76%                |
+|------------------------------------------|------------------|-----------------------|\
+| `resnet18_fashionmnist_original.pth`     | None             | 95.39%                |\
+| `resnet18_fashionmnist_forget0.pth`      | T-shirt/top      | 96.54%                |\
+| `resnet18_fashionmnist_forget1.pth`      | Trouser          | 95.07%                |\
+| `resnet18_fashionmnist_forget2.pth`      | Pullover         | 95.99%                |\
+| `resnet18_fashionmnist_forget3.pth`      | Dress            | 95.69%                |\
+| `resnet18_fashionmnist_forget4.pth`      | Coat             | 95.92%                |\
+| `resnet18_fashionmnist_forget5.pth`      | Sandal           | 94.88%                |\
+| `resnet18_fashionmnist_forget6.pth`      | Shirt            | 97.76%                |\
+| `resnet18_fashionmnist_forget7.pth`      | Sneaker          | 95.37%                |\
+| `resnet18_fashionmnist_forget8.pth`      | Bag              | 94.76%                |\
 | `resnet18_fashionmnist_forget9.pth`      | Ankle boot       | 95.33%                |
 
 ## Training Details
@@ -58,16 +73,12 @@ This repository contains ResNet-18 models retrained on the FashionMNIST dataset
   - Evaluate retained-class performance on the remaining 9 classes’ test samples.  
   - (Optional) Evaluate excluded-class accuracy separately to confirm near-zero performance.
 
-## Related Work
-This model is part of the research conducted using the [Machine Unlearning Comparator](https://github.com/gnueaj/Machine-Unlearning-Comparator). The tool was developed to compare various machine unlearning methods and their effects on models.
-
 ## Uses
 ## Direct Use
-These models can be directly used for evaluating the effect of excluding specific classes from the CIFAR-10 dataset during training.
+These models can be directly used for evaluating the effect of excluding specific classes from the FashionMNIST dataset during training.
 
 ## Out-of-Scope Use
-The models are not suitable for tasks requiring general-purpose image classification beyond the CIFAR-10 dataset.
-
+The models are not suitable for tasks requiring general-purpose image classification beyond the FashionMNIST dataset.
 
 ### Model Definition Example
 ```python
@@ -82,3 +93,4 @@ def ResNet18_FashionMNIST(num_classes=10, in_channels=1):
     model.maxpool = nn.Identity()
     model.fc = nn.Linear(model.fc.in_features, num_classes)
     return model
+```