ドライブレコーダーの Scene Text Recognitionにおける Multi-task Learning

2. 2 Proprietary and Confidential ©2017 JapanTaxi, Inc. All Rights Reserved • 名前：⾼橋⽂彦 • 略歴 • 2015年4⽉ヤフー株式会社⼊社 • 2018年8⽉ JapanTaxi株式会社⼊社 • 領域 • 画像処理、⾃然⾔語処理 • 過去の主な仕事 • 配⾞アプリのお迎え時間予測 • ECサイトの検索精度改善 • その他：PM、論⽂書いたり • 趣味 • ボードゲーム、IoTガジェット

14. 14 Proprietary and Confidential ©2017 JapanTaxi, Inc. All Rights Reserved 課題：学習データを収集するのが⼤変 100.0% 6.9% 2.7% 地図を利⽤しても3%程度しか学習データが集まらない Detection アノテーション Recognition アノテーション Sampling with location Sampled images Detected images Recognized images 地図とGPSの情報を使って写っている可能性の⾼い画像を抽出

15. 15 Proprietary and Confidential ©2017 JapanTaxi, Inc. All Rights Reserved 課題：学習データを収集するのが⼤変 100.0% 2.6% 0.2% 全体の0.2%程度しか学習データが集まらない Random Sampling ⾃動 Detection Recognition アノテーション Detected images Recognized images

18. 18 Proprietary and Confidential ©2017 JapanTaxi, Inc. All Rights Reserved ⼀般的なSean Text Recognitionのモデル CNN Layers Bi-LSTM Layers Recognition Layer --1222--22-- 122 • CTC(Connectionist Temporal Classification)で⽂字列を予測 • ⾳声認識でも使われる系列予測の⼿法 [An End-to-End Trainable Neural Network for Image-based Sequence Recognition and Its Application to Scene Text Recognition, Baoguang Shi et al., IEEE Trans. Pattern Anal. Mach. Intell. 2017] CNN-RNN Model (baseline)

19. 19 Proprietary and Confidential ©2017 JapanTaxi, Inc. All Rights Reserved 関連研究：Multi-task Learning [An Overview of Multi-Task Learning in Deep Neural Networks, Sebastian Ruder, 2017] 関連する複数のタスクを同時に学習させることで、予測精度を向上させる

21. 21 Proprietary and Confidential ©2017 JapanTaxi, Inc. All Rights Reserved Multi-task Learning of Scene Text Recognition CNN Layers Bi-LSTM Layers Recognition Layer 122 • Classificationは数字が写っている場合1.0、写ってない場合0.0 • RecognitionとClassificationの両⽅のlossからBack Propagation 1.0 数字が写っている画像の学習 Classification Layer

22. 22 Proprietary and Confidential ©2017 JapanTaxi, Inc. All Rights Reserved Multi-task Learning of Scene Text Recognition CNN Layers Bi-LSTM Layers Recognition Layer • ClassificationのlossからのみBack Propagation • ClassificationのLoss functionはMean Squared Error 0.0 数字が写っていない画像の学習 Classification Layer

25. 25 Proprietary and Confidential ©2017 JapanTaxi, Inc. All Rights Reserved 設定 • training set と validation setは2:8になるように時間軸で分割 • データ数に偏りがあるため、Classificationのサンプルごとの重みを画像枚数の割合で設定 • 数字あり(Labeled)：U/(U+L) • 数字なし（Unlabeled）：L/(U+L) 擬似画像数字あり画像 5,280,000 数字なし画像 2,640,000 実画像数字あり画像 402 数字なし画像 2,274

26. 26 Proprietary and Confidential ©2017 JapanTaxi, Inc. All Rights Reserved 精度評価結果 Text Recognition Classification Accuracy Normalized Edit Distance Precision Recall Single-task Model 0.3088 0.3725 - - Multi-task Model 0.6324 0.1618 0.8824 0.5217 • Multi-task Modelの⽅がRecognitionのAccuracyが⾼い • 実⽤では… • 画像サイズの⼤きさ画像と精度に相関があり、⾞両が近づけば⾼い精度で認識可能 • フレーム間補完を⾏う

27. 27 Proprietary and Confidential ©2017 JapanTaxi, Inc. All Rights Reserved 学習データ量と精度の関係 0.000 0.100 0.200 0.300 0.400 0.500 0.600 0.700 0.0% 20.0% 40.0% 60.0% 80.0% 100.0% Multi-task Model Single-task Model • Multi-taskは少量のデータでも⽴ち上がりが早く、すでに鈍りが⾒えている • Single-taskは学習データを増やすと精度が線形に向上

30. 30 Proprietary and Confidential ©2017 JapanTaxi, Inc. All Rights Reserved 背景：ドライブレコーダーの動画からガソリンスタンドの料⾦を⾃動認識課題：学習データを取集するのが⼤変結論： • 数字じゃない形（データが⼤量）も学習することで少ないデータで精度が向上 • 学習データが少ない時にもMulti-task Learningは有効 • 実⽤上、数字が書かれているかどうかの判定は必要なので、１つのモデルに詰め込めたのは⼀⽯⼆⿃だった

31. 〒102-0094 東京都千代⽥区紀尾井町3-12 3-12 Kioicho Chiyoda-ku, Tokyo 102-0094 Japan TEL 03-6265-6265 FAX 03-3239-8115 www.japantaxi.co.jp ⽂章·画像等の内容の無断転載及び複製等の⾏為はご遠慮ください。 Proprietary and Confidential ©2017 JapanTaxi, Inc. All Rights Reserved

ドライブレコーダーの Scene Text Recognitionにおける Multi-task Learning

Estás leyendo una previsualización.

Eche un vistazo a continuación

ドライブレコーダーの Scene Text Recognitionにおける Multi-task Learning

Más Contenido Relacionado

Similares a ドライブレコーダーの Scene Text Recognitionにおける Multi-task Learning (20)

Más reciente (20)