Learning multi-task communication with message passing

Pengfei Liu, Jie Fu*, Yue Dong*, Xipeng Qiu, Jackie Chi Kit Cheung

Fig. Different topology structures for multi-task learning: flat, hierarchical and graph structures. Each blue circle represents a task (A, B, C, D), while the red box denotes a virtual node, which stores shared information and facilitates communication


Case Study

Fig. : Illustrations of the interpretable patterns captured by the shared layer under different tasks.