arXiv reaDer
肝臓セグメンテーションのスケジューリング手法:ReduceLRonPlateauとOneCycleLR
Scheduling Techniques for Liver Segmentation: ReduceLRonPlateau Vs OneCycleLR
機械学習とコンピュータービジョン技術は、生物医学分野を含む多くの分野に影響を与えてきました。このホワイトペーパーの目的は、ReduceLRonPlateauに対して新しく考案されたOneCycleLRに焦点を当て、トレーニングプロセス全体を通じて、肝臓セグメンテーションタスクの学習率(LR)を操作する際のスケジューラの重要な概念を調査することです。 2018年に公開され、Medical Segmentation Decathlon Challengeの主催者によって作成された、Task 8 Hepatic Vessel(MSDC-T8)と呼ばれるデータセットが、テストと検証に使用されました。最大エポック数が同じ(75)で、5分割交差検定の平均である報告された結果は、OneCycleLRと比較した場合、検証セットで同様またはさらに優れた損失スコアを維持しながら、ReduceLRonPlateauがより速く収束することを示しています。ピークLRが発生するエポックは、超収束機能を観察できるように、OneCycleLRではおそらく早期に作成する必要があります。さらに、全体的な結果は、このデータセットの肝臓マスクを公開した研究者による最新の結果を上回っています。結論として、両方のスケジューラーは、医療セグメンテーションの課題、特にMSDC-T8データセットに適しており、最小限のエポックで検証損失を迅速に収束するために自信を持って使用できます。
Machine learning and computer vision techniques have influenced many fields including the biomedical one. The aim of this paper is to investigate the important concept of schedulers in manipulating the learning rate (LR), for the liver segmentation task, throughout the training process, focusing on the newly devised OneCycleLR against the ReduceLRonPlateau. A dataset, published in 2018 and produced by the Medical Segmentation Decathlon Challenge organizers, called Task 8 Hepatic Vessel (MSDC-T8) has been used for testing and validation. The reported results that have the same number of maximum epochs (75), and are the average of 5-fold cross-validation, indicate that ReduceLRonPlateau converges faster while maintaining a similar or even better loss score on the validation set when compared to OneCycleLR. The epoch at which the peak LR occurs perhaps should be made early for the OneCycleLR such that the super-convergence feature can be observed. Moreover, the overall results outperform the state-of-the-art results from the researchers who published the liver masks for this dataset. To conclude, both schedulers are suitable for medical segmentation challenges, especially the MSDC-T8 dataset, and can be used confidently in rapidly converging the validation loss with a minimal number of epochs.
updated: Sun Feb 13 2022 17:52:49 GMT+0000 (UTC)
published: Sun Feb 13 2022 17:52:49 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト