arXiv reaDer
YOLOrtho -- 歯の数え上げと歯の病気の検出のための統合フレームワーク
YOLOrtho -- A Unified Framework for Teeth Enumeration and Dental Disease Detection
パノラマ X 線画像を通じて歯科疾患を検出することは、歯科医にとって標準的な手順です。通常、歯科医は病気を特定し、感染した歯を見つける必要があります。この 2 段階の手順を採用した多数の機械学習モデルが開発されていますが、歯とそれに関連する疾患を同時に識別できるエンドツーエンドのモデルはありませんでした。このギャップを埋めるために、私たちは歯の列挙と歯科疾患の検出のための統合フレームワークである YOLOrtho を開発しました。私たちは、3 つの異なるタイプの注釈付きデータで構成される Dentex Challenge 2023 データに基づいてモデルを開発しました。最初の部分には四分円、二番目の部分には四分円と列挙、そして三番目の部分には四分円、列挙、疾患というラベルが付いています。検出をさらに向上させるために、Tufts Dental の公開データセットを利用します。データを最大限に活用し、歯の検出と病気の特定の両方を同時に学習するために、病気を対応する歯に付随する属性として定式化します。歯の列挙における位置関係の性質により、モデルの畳み込み層を CoordConv に置き換えて、モデルにより多くの位置情報を提供します。また、モデル アーキテクチャを調整し、大きなオブジェクトの検出を優先して FPN にアップサンプリング レイヤーを 1 つ追加します。最後に、線形和の割り当てに基づいて歯の列挙を修正する歯のレイアウトの後処理戦略を提案します。実験の結果、私たちのモデルは大規模な拡散ベースのモデルを上回っていることがわかりました。
Detecting dental diseases through panoramic X-rays images is a standard procedure for dentists. Normally, a dentist need to identify diseases and find the infected teeth. While numerous machine learning models adopting this two-step procedure have been developed, there has not been an end-to-end model that can identify teeth and their associated diseases at the same time. To fill the gap, we develop YOLOrtho, a unified framework for teeth enumeration and dental disease detection. We develop our model on Dentex Challenge 2023 data, which consists of three distinct types of annotated data. The first part is labeled with quadrant, and the second part is labeled with quadrant and enumeration and the third part is labeled with quadrant, enumeration and disease. To further improve detection, we make use of Tufts Dental public dataset. To fully utilize the data and learn both teeth detection and disease identification simultaneously, we formulate diseases as attributes attached to their corresponding teeth. Due to the nature of position relation in teeth enumeration, We replace convolution layer with CoordConv in our model to provide more position information for the model. We also adjust the model architecture and insert one more upsampling layer in FPN in favor of large object detection. Finally, we propose a post-process strategy for teeth layout that corrects teeth enumeration based on linear sum assignment. Results from experiments show that our model exceeds large Diffusion-based model.
updated: Fri Aug 11 2023 06:54:55 GMT+0000 (UTC)
published: Fri Aug 11 2023 06:54:55 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アソシエイト