HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

You Only Learn One Representation: Unified Network for Multiple Tasks

Chien-Yao Wang I-Hau Yeh Hong-Yuan Mark Liao

You Only Learn One Representation: Unified Network for Multiple Tasks

Abstract

People ``understand'' the world via vision, hearing, tactile, and also the past experience. Human experience can be learned through normal learning (we call it explicit knowledge), or subconsciously (we call it implicit knowledge). These experiences learned through normal learning or subconsciously will be encoded and stored in the brain. Using these abundant experience as a huge database, human beings can effectively process data, even they were unseen beforehand. In this paper, we propose a unified network to encode implicit knowledge and explicit knowledge together, just like the human brain can learn knowledge from normal learning as well as subconsciousness learning. The unified network can generate a unified representation to simultaneously serve various tasks. We can perform kernel space alignment, prediction refinement, and multi-task learning in a convolutional neural network. The results demonstrate that when implicit knowledge is introduced into the neural network, it benefits the performance of all tasks. We further analyze the implicit representation learnt from the proposed unified network, and it shows great capability on catching the physical meaning of different tasks. The source code of this work is at : https://github.com/WongKinYiu/yolor.

Code Repositories

totoadel/Minst
pytorch
Mentioned in GitHub
DataXujing/YOLOR-
pytorch
Mentioned in GitHub
SnailDev/github-hot-hub
pytorch
Mentioned in GitHub
alexeyab/yolor
pytorch
Mentioned in GitHub
WongKinYiu/yolor
Official
pytorch
Mentioned in GitHub
lonnyzhang423/github-hot-hub
pytorch
Mentioned in GitHub
rjlallana/YOLO
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
object-detection-on-cocoYOLOR-D6 (1280, single-scale, 30 fps)
AP50: 73.3
AP75: 60.6
box mAP: 55.4
object-detection-on-coco-minivalYOLOR-P6 (1280, single-scale, 72 fps)
AP50: 70.6
AP75: 57.4
APL: 65.2
APM: 57.3
APS: 37.4
object-detection-on-coco-minivalYOLOR-D6 (1280, single-scale, 31 fps)
AP50: 73.5
AP75: 60.6
APL: 68.7
APM: 60.1
APS: 40.4
real-time-object-detection-on-cocoYOLOR-E6
FPS (V100, b=1): 37
box AP: 54.8
real-time-object-detection-on-cocoYOLOR-P6
FPS (V100, b=1): 49
box AP: 52.6
real-time-object-detection-on-cocoYOLOR-W6
FPS (V100, b=1): 47
box AP: 54.1
real-time-object-detection-on-cocoYOLOR-P6D
FPS (V100, b=1): 49
box AP: 53
real-time-object-detection-on-cocoYOLOR-D6
FPS (V100, b=1): 30
box AP: 55.4

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
You Only Learn One Representation: Unified Network for Multiple Tasks | Papers | HyperAI