损失函数比较结果数据集

数据集获取（Figshare）

• 源站标题：Results of the comparison of the loss functions.
• DOI：10.1371/journal.pone.0313277.t005
• 发布日期：2025-01-30
• 文件数：1 个
• 文件总体积：5.63 KB
• 源站页面：https://plos.figshare.com/articles/dataset/Results_of_the_comparison_of_the_loss_functions_/28316002

命令行下载

curl -L -o "Table 5.xls" "https://ndownloader.figshare.com/files/52045168"

文件清单

• Table 5.xls（5.63 KB）

文件总体积 5.63 KB（源站 API 实测）。

数据加载示例（图像类）

from PIL import Image
import glob, os

files = (glob.glob(os.path.join(path, "**", "*.png"), recursive=True)
       + glob.glob(os.path.join(path, "**", "*.jpg"), recursive=True)
       + glob.glob(os.path.join(path, "**", "*.tif"), recursive=True))
print("图像文件数:", len(files))
img = Image.open(files[0]); print("尺寸/模式:", img.size, img.mode)

# torchvision Dataset 方式:
# from torchvision import datasets
# ds = datasets.ImageFolder(path)  # 要求 子目录=类别

目录组织与标注格式以源站说明和下载后实际文件为准。

数据说明

The incidence of acute myeloid leukemia (AML) is increasing annually, and timely diagnostic and treatments can substantially improve patient survival rates. AML typing traditionally relies on manual microscopy for classifying and counting myeloid cells, which is time-consuming, laborious, and subjective. Therefore, developing a reliable automated model for myeloid cell classification is imperative. This study evaluated the performance of five widely-used classification models on the largest publicly available bone marrow cell dataset (BM). However, the accuracy of the classification model is significantly affected by the imbalance in the distribution of bone marrow cell types. To address this issue, this study analyzed five different Loss functions and seven different attention mechanisms. When the classification models is chosen, Swin Transformer V2 was found to perform the best. However, the lightweight model RegNetX-3.2gf had significantly fewer parameters and a significantly faster inference speed than Swin Transformer V2, and its F1 Score was only 0.032 lower than that of Swin Transformer V2. Accordingly, RegNetX-3.2gf is strongly recommended for practical applications. During the evaluation of Loss function and attention mechanism, the Cost-Sensitive Loss Function (CS) and the channel attention mechanism Squeeze-and-Excitation Networks (SE) demonstrated superior performance. The optimal model (RegNetX-3.2gf + CS + SE) achieved an average precision of 68.183%, an average recall of 63.722%, and an average F1 Score of 65.155%. This model exhibited significantly improved performance compared to the original dataset results, achieving an enhancement of 17.183% in precision and 10.655% in the F1 Score. Finally, the class activation maps demonstrate that our model focused on the cells themselves, especially on the nucleus when making classifications. It proved that our model was reliable. This study provided an important reference for the study of bone marrow cell classification and a practical application of the model, promoting the development of the intelligent classification of AML.

许可

CC BY 4.0

引用

学术使用请引用 DOI 10.1371/journal.pone.0313277.t005（source: https://plos.figshare.com/articles/dataset/Results_of_the_comparison_of_the_loss_functions_/28316002）。