TranSalNet: Towards perceptually relevant visual saliency prediction

7 Oct 2021  ·  Jianxun Lou, Hanhe Lin, David Marshall, Dietmar Saupe, Hantao Liu ·

Visual saliency prediction using transformers - Convolutional neural networks (CNNs) have significantly advanced computational modelling for saliency prediction. However, accurately simulating the mechanisms of visual attention in the human cortex remains an academic challenge. It is critical to integrate properties of human vision into the design of CNN architectures, leading to perceptually more relevant saliency prediction. Due to the inherent inductive biases of CNN architectures, there is a lack of sufficient long-range contextual encoding capacity. This hinders CNN-based saliency models from capturing properties that emulate viewing behaviour of humans. Transformers have shown great potential in encoding long-range information by leveraging the self-attention mechanism. In this paper, we propose a novel saliency model that integrates transformer components to CNNs to capture the long-range contextual visual information. Experimental results show that the transformers provide added value to saliency prediction, enhancing its perceptual relevance in the performance. Our proposed saliency model using transformers has achieved superior results on public benchmarks and competitions for saliency prediction models. The source code of our proposed saliency model TranSalNet is available at:

PDF Abstract


Results from the Paper

Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Saliency Prediction MIT300 TranSalNet CC 0.807 # 1
SIM 0.6895 # 1
NSS 2.4134 # 1
sAUC 0.7467 # 1
AUC-Judd 0.8734 # 1
KLD 1.0141 # 1
Saliency Prediction SALECI Transalnet KL 0.873 # 4
Saliency Prediction SALICON TranSalNet CC 0.907 # 2
SIM 0.803 # 1
NSS 2.014 # 1
sAUC 0.747 # 1
AUC 0.868 # 2
KLD 0.373 # 1