Regularization for Unsupervised Learning of Optical Flow

Regularization is an important technique for training deep neural networks. In this paper, we propose a novel shared-weight teacher–student strategy and a content-aware regularization (CAR) module. Based on a tiny, learnable, content-aware mask, CAR is randomly applied to some channels in the convol...

Full description

Bibliographic Details
Main Authors: Libo Long, Jochen Lang
Format: Article
Language:English
Published: MDPI AG 2023-04-01
Series:Sensors
Subjects:
Online Access:https://www.mdpi.com/1424-8220/23/8/4080
_version_ 1797603488459390976
author Libo Long
Jochen Lang
author_facet Libo Long
Jochen Lang
author_sort Libo Long
collection DOAJ
description Regularization is an important technique for training deep neural networks. In this paper, we propose a novel shared-weight teacher–student strategy and a content-aware regularization (CAR) module. Based on a tiny, learnable, content-aware mask, CAR is randomly applied to some channels in the convolutional layers during training to be able to guide predictions in a shared-weight teacher–student strategy. CAR prevents motion estimation methods in unsupervised learning from co-adaptation. Extensive experiments on optical flow and scene flow estimation show that our method significantly improves on the performance of the original networks and surpasses other popular regularization methods. The method also surpasses all variants with similar architectures and the supervised PWC-Net on MPI-Sintel and on KITTI. Our method shows strong cross-dataset generalization, i.e., our method solely trained on MPI-Sintel outperforms a similarly trained supervised PWC-Net by 27.9% and 32.9% on KITTI, respectively. Our method uses fewer parameters and less computation, and has faster inference times than the original PWC-Net.
first_indexed 2024-03-11T04:32:50Z
format Article
id doaj.art-3f6b7226f7864cb2bac27377618e41be
institution Directory Open Access Journal
issn 1424-8220
language English
last_indexed 2024-03-11T04:32:50Z
publishDate 2023-04-01
publisher MDPI AG
record_format Article
series Sensors
spelling doaj.art-3f6b7226f7864cb2bac27377618e41be2023-11-17T21:18:46ZengMDPI AGSensors1424-82202023-04-01238408010.3390/s23084080Regularization for Unsupervised Learning of Optical FlowLibo Long0Jochen Lang1Faculty of Engineering, University of Ottawa, Ottawa, ON K1N 6N5, CanadaFaculty of Engineering, University of Ottawa, Ottawa, ON K1N 6N5, CanadaRegularization is an important technique for training deep neural networks. In this paper, we propose a novel shared-weight teacher–student strategy and a content-aware regularization (CAR) module. Based on a tiny, learnable, content-aware mask, CAR is randomly applied to some channels in the convolutional layers during training to be able to guide predictions in a shared-weight teacher–student strategy. CAR prevents motion estimation methods in unsupervised learning from co-adaptation. Extensive experiments on optical flow and scene flow estimation show that our method significantly improves on the performance of the original networks and surpasses other popular regularization methods. The method also surpasses all variants with similar architectures and the supervised PWC-Net on MPI-Sintel and on KITTI. Our method shows strong cross-dataset generalization, i.e., our method solely trained on MPI-Sintel outperforms a similarly trained supervised PWC-Net by 27.9% and 32.9% on KITTI, respectively. Our method uses fewer parameters and less computation, and has faster inference times than the original PWC-Net.https://www.mdpi.com/1424-8220/23/8/4080self-supervised trainingteacher–student learningregularizationoptical flowscene flow
spellingShingle Libo Long
Jochen Lang
Regularization for Unsupervised Learning of Optical Flow
Sensors
self-supervised training
teacher–student learning
regularization
optical flow
scene flow
title Regularization for Unsupervised Learning of Optical Flow
title_full Regularization for Unsupervised Learning of Optical Flow
title_fullStr Regularization for Unsupervised Learning of Optical Flow
title_full_unstemmed Regularization for Unsupervised Learning of Optical Flow
title_short Regularization for Unsupervised Learning of Optical Flow
title_sort regularization for unsupervised learning of optical flow
topic self-supervised training
teacher–student learning
regularization
optical flow
scene flow
url https://www.mdpi.com/1424-8220/23/8/4080
work_keys_str_mv AT libolong regularizationforunsupervisedlearningofopticalflow
AT jochenlang regularizationforunsupervisedlearningofopticalflow