File size: 3,371 Bytes
94b2815
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
bba3774
94b2815
 
bba3774
94b2815
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
---
title: Continual learning results (%) on data separated by region groups
license: mit  
task_categories:
- image-classification
tags:
- continual-learning
- resnet
- vit
- vit-hybrid        
- swin-transformer  
---

# Continual Learning Results (Region Groups)

This repository contains the weights and experimental results of continual learning fine-tuning using several mainstream vision backbones. The methods evaluated include FineTune (Cross Entropy), EWC, VCL, and FIP.

## πŸ“‚ Directory Structure

```text
.
β”œβ”€β”€ FineTune_Cross_Entropy/
β”‚   β”œβ”€β”€ model/
β”‚   β”‚   β”œβ”€β”€ config.json              # Model configuration and hyperparameters
β”‚   β”‚   └── pytorch_model.bin        # Model weights file
β”‚   └── result/
β”‚       β”œβ”€β”€ misclassified_images.csv # Records of misclassified images
β”‚       β”œβ”€β”€ test_evaluation.csv      # Evaluation metrics and test results
β”‚       β”œβ”€β”€ training_time.csv        # Logged training duration per group (hours)
β”‚       └── training_metrics_set1.csv ~ training_metrics_set7.csv # Training dynamics and output logs
β”œβ”€β”€ EWC/
β”‚   β”œβ”€β”€ model/
β”‚   β”‚   β”œβ”€β”€ config.json
β”‚   β”‚   └── pytorch_model.bin
β”‚   └── result/
β”‚       β”œβ”€β”€ misclassified_images.csv
β”‚       β”œβ”€β”€ test_evaluation.csv
β”‚       β”œβ”€β”€ training_time.csv
β”‚       └── training_metrics_set1.csv ~ training_metrics_set7.csv
β”œβ”€β”€ VCL/
β”‚   β”œβ”€β”€ model/
β”‚   β”‚   β”œβ”€β”€ config.json
β”‚   β”‚   └── pytorch_model.bin
β”‚   └── result/
β”‚       β”œβ”€β”€ misclassified_images.csv
β”‚       β”œβ”€β”€ test_evaluation.csv
β”‚       β”œβ”€β”€ training_time.csv
β”‚       └── training_metrics_set1.csv ~ training_metrics_set7.csv
└── FIP/
    β”œβ”€β”€ model/
    β”‚   β”œβ”€β”€ config.json
    β”‚   └── pytorch_model.bin
    └── result/
        β”œβ”€β”€ misclassified_images.csv
        β”œβ”€β”€ test_evaluation.csv
        β”œβ”€β”€ training_time.csv
        └── training_metrics_set1.csv ~ training_metrics_set7.csv
```

## πŸš€ Usage

You can download the model weights for a specific continual learning method using the `huggingface_hub` SDK:

```python
from huggingface_hub import hf_hub_download

# Example: Download Swin Transformer model weights for the VCL method
model_weight_path = hf_hub_download(
    repo_id="Dataset-and-Model/continual-learning-region-groups",
    filename="Swin Transformer/VCL/model/pytorch_model.bin",
    repo_type="model"
)

print(f"Weights successfully downloaded to: {model_weight_path}")
```

## πŸ—οΈ Base Models

The fine-tuning and continual learning experiments in this study are built upon the following pre-trained models provided by Hugging Face:
* **ResNet-34** (e.g., `microsoft/resnet-34`)
* **Vision Transformer (ViT)** (e.g., `google/vit-large-patch32-384`)
* **ViT Hybrid** (e.g., `google/vit-hybrid-base-bit-384`)
* **SWIN Transformer** (e.g., `microsoft/swin-large-patch4-window12-384`)

## πŸ“œ License

These fine-tuned model weights are derived from the open-source architectures mentioned above and are released under the **MIT License**. You are free to use, modify, and distribute these weights in both academic research and commercial applications, provided that proper attribution to the original authors is maintained.