Replace link to GitHub with Code Ocean, add new paper citation

#5
Files changed (1) hide show
  1. README.md +16 -3
README.md CHANGED
@@ -34,7 +34,7 @@ The second model version ([2371ff8](https://huggingface.co/RikeB/MaxViT_butterfl
34
 
35
  Training was parallelized using the Pytorch DDP framework and the Hugging Face Accelerate library.
36
  More information on model training can be found in the publications below.
37
- Scripts are availble on [GitHub](https://github.com/AndiLindner/butterfly_identification).
38
 
39
  ### Training Data
40
 
@@ -47,7 +47,7 @@ Species with less than 50 images were excluded from training. The final dataset
47
  The first model version was trained in the context of a data paper in which the butterfly and moth images dataset it was trained on was published:
48
 
49
  ```bibtex
50
- @Article{Barkmannetal2025a,
51
  author={Barkmann, Friederike
52
  and Lindner, Andreas
53
  and W{\"u}rflinger, Ronald
@@ -68,4 +68,17 @@ url={https://doi.org/10.1038/s41597-025-05708-z}
68
  }
69
  ```
70
 
71
- Another publication is in preparation.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
34
 
35
  Training was parallelized using the Pytorch DDP framework and the Hugging Face Accelerate library.
36
  More information on model training can be found in the publications below.
37
+ Scripts are availble on [Code Ocean](https://doi.org/10.24433/CO.7366644.v1).
38
 
39
  ### Training Data
40
 
 
47
  The first model version was trained in the context of a data paper in which the butterfly and moth images dataset it was trained on was published:
48
 
49
  ```bibtex
50
+ @Article{Barkmannetal2025,
51
  author={Barkmann, Friederike
52
  and Lindner, Andreas
53
  and W{\"u}rflinger, Ronald
 
68
  }
69
  ```
70
 
71
+ The second model was obtained by a more thorough training described in the following paper:
72
+
73
+ ```bibtex
74
+ @article{Barkmannetal2026,
75
+ author = {Barkmann, Friederike and Lindner, Andreas and R{\"u}disser, Johannes},
76
+ title = {Comparing deep learning models for butterfly and moth ({Lepidoptera}) species identification},
77
+ journal = {Insect Conservation and Diversity},
78
+ year = {2026},
79
+ issn = {1752-458X},
80
+ doi = {10.1111/icad.70123},
81
+ url = {https://doi.org/10.1111/icad.70123},
82
+ abstract = {The number of image-based species records has increased exponentially in recent years. Deep learning models offer opportunities for efficient data processing, especially for species identification. Evaluating models on comprehensive datasets can provide insights into their suitability for species identification. Based on a high-quality dataset with more than 500,000 images, we compared 40 models from 12 classes on the classification of 162 butterfly and moth species, and further analysed the results of the best model. The dataset covers most Austrian butterfly species, but a few difficult-to-identify species are not included or were grouped. Multiple models reached validation accuracies >97\%, with larger models generally performing better, but there were some notable exceptions. The MaxViT-tiny model achieved the highest accuracy despite being one of the smaller models. After further hyperparameter optimisation, the performance of the MaxViT was increased to a Top-1 accuracy of 98.14\% and a Top-5 accuracy of 99.53\% on test data. Although methods to correct for class imbalance were applied, recall and precision were generally lower for species with fewer images in the dataset. Our results emphasise the high potential of deep learning for species identification for the species analysed in this study. The results aid projects with limited computational capacities in choosing an efficient and appropriate model. Additional data processing steps, such as applying quality criteria to exclude model predictions with low confidence and expert assessments for difficult-to-determine species, can substantially increase the quality of identification.}
83
+ }
84
+ ```