STENet: Superpixel Token Enhancing Network for RGB-D Salient Object Detection
Organizations: School of Communication and Information Engineering, Shanghai University, Shanghai 200444, China · Yunnan Key Laboratory of Service Computing, Yunnan University of Finance and Economics, Kunming 650000, China · Innovation Academy for Microsatellites of Chinese Academy of Science, Shanghai 201203, China · Institute for Urban Governance, Shanghai University, Shanghai 200444, China
Abstract
Transformer-based methods for RGB-D Salient Object Detection (SOD) have gained significant interest, owing to the transformer's exceptional capacity to capture long-range pixel dependencies. Nevertheless, current RGB-D SOD methods face challenges, such as the quadratic complexity of the attention mechanism and the limited local detail extraction. To overcome these limitations, we propose a novel Superpixel Token Enhancing Network (STENet), which introduces superpixels into cross-modal interaction. STENet follows the two-stream encoder-decoder structure. Its cores are two tailored superpixel-driven cross-modal interaction modules, responsible for global and local feature enhancement. Specifically, we update the superpixel generation method by expanding the neighborhood range of each superpixel, allowing for flexible transformation between pixels and superpixels. With the updated superpixel generation method, we first propose the Superpixel Attention Global Enhancing Module to model the global pixel-to-superpixel relationship rather than the traditional global pixel-to-pixel relationship, which can capture region-level information and reduce computational complexity. We also propose the Superpixel Attention Local Refining Module, which leverages pixel similarity within superpixels to filter out a subset of pixels (i.e., local pixels) and then performs feature enhancement on these local pixels, thereby capturing concerned local details. Furthermore, we fuse the globally and locally enhanced features along with the cross-scale features to achieve comprehensive feature representation. Experiments on seven RGB-D SOD datasets reveal that our STENet achieves competitive performance compared to state-of-the-art methods. The code and results of our method are available at https://github.com/Mark9010/STENet.
Figures & tables
| Model | Publisher | Year | Backbone | #Param | FLOPs | NJUD [ 53 ] | NLPR [ 54 ] | DUT-RGBD [ 55 ] | ||||||||||||
| (M)↓ | (G)↓ | ↓ | ↑ | ↑ | ↑ | ↑ | ↓ | ↑ | ↑ | ↑ | ↑ | ↓ | ↑ | ↑ | ↑ | ↑ | ||||
| DSNet [ 56 ] | TIP | 2021 | Res50 | - | - | .034 | .922 | .947 | .921 | .898 | .024 | .917 | .956 | .926 | .886 | .079 | .821 | .886 | .841 | .774 |
| DCFNet [ 30 ] | CVPR | 2021 | Res50 | 108.5 | 107.8 | .035 | .915 | .944 | .911 | .893 | .021 | .912 | .959 | .923 | .892 | .071 | .835 | .891 | .836 | .766 |
| CIRNet [ 57 ] | TIP | 2022 | Res50 | 103.1 | 22.5 | .035 | .927 | .943 | .925 | .895 | .023 | .924 | .955 | .933 | .889 | .031 | .940 | .958 | .932 | .904 |
| C2DFNet [ 38 ] | TMM | 2022 | Res50 | 47.5 | 22.1 | .038 | .909 | .941 | .907 | .885 | .021 | .916 | .961 | .927 | .897 | . 026 | .943 | . 963 | .933 | . 918 |
| DIGRNet [ 58 ] | TMM | 2022 | Res50 | 201.8 | 68.2 | . 028 | . 936 | . 954 | . 933 | .909 | .023 | .928 | .956 | . 935 | .895 | .033 | .938 | .951 | .926 | .898 |
| Model | Backbone | LFSD [ 65 ] | SIP [ 66 ] | STEREO [ 67 ] | SSD [ 68 ] | ||||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| ↓ | ↑ | ↑ | ↑ | ↑ | ↓ | ↑ | ↑ | ↑ | ↑ | ↓ | ↑ | ↑ | ↑ | ↑ | ↓ | ↑ | ↑ | ↑ | ↑ | ||
| DSNet [ 56 ] | Res50 | .069 | .858 | .905 | .868 | .826 | .052 | .881 | .920 | .876 | .840 | .036 | .911 | .947 | .915 | .882 | .045 | . 878 | .923 | . 885 | .838 |
| DCFNet [ 30 ] | Res50 | .075 | .841 | .883 | .841 | .805 | .051 | .884 | .921 | .875 | .848 | .039 | .901 | .941 | .901 | .875 | .049 | .851 | .905 | .864 | .814 |
| CIRNet [ 57 ] | Res50 | .068 | . 883 | .904 | .875 | .838 | .052 | .896 | .918 | .888 | .848 | .039 | .913 | .939 | .915 | .872 | .049 | .864 | .911 | .878 | .816 |
| C2DFNet [ 38 ] | Res50 | .065 | .867 | .902 | .863 | .835 | .052 | .877 | .918 | .871 | .841 | .038 | .897 | .942 | .902 | .871 | .047 | .860 | .920 | .872 | .827 |
| DIGRNet [ 58 ] | Res50 | .067 | .865 | .906 | .873 | .828 | .053 | .897 | .918 | .885 | .849 | .038 | . 914 | .943 | . 916 | .877 | .053 | .846 | .898 | .866 | .804 |
| NJUD [ 53 ] | LFSD [ 65 ] | SIP [ 66 ] | Average | |||||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Model | ↓ | ↑ | ↑ | ↑ | ↑ | ↓ | ↑ | ↑ | ↑ | ↑ | ↓ | ↑ | ↑ | ↑ | ↑ | ↓ | ↑ | ↑ | ↑ | ↑ |
| Baseline | .029 | .930 | .954 | .926 | .913 | .057 | .880 | .914 | .877 | .848 | .036 | .920 | .945 | .908 | .894 | .041 | .910 | .938 | .904 | .885 |
| +SAGEM | .026 | .936 | .963 | .927 | .925 | .050 | .884 | .917 | .880 | .861 | .037 | .924 | .941 | .901 | .899 | .038 | .915 | .940 | .903 | .895 |
| +SALRM | .025 | .935 | .961 | .927 | .925 | .051 | .879 | .917 | .879 | .857 | .036 | .924 | .944 | .900 | .900 | .037 | .913 | .941 | .902 | .894 |
| +SAGEM+SALRM (Ours) | .023 | .941 | .967 | .932 | .931 | .048 | .891 | .928 | .890 | .873 | .032 | .931 | .952 | .910 | .911 | .034 | .921 | .949 | .911 | .905 |
| +CAF+SALRM | .023 | .936 | .963 | .929 | .926 | .047 | .891 | .929 | .891 | .873 | .036 | .920 | .943 | .902 | .897 | .035 | .916 | .945 | .907 | .899 |
| NJUD [ 53 ] | LFSD [ 65 ] | SIP [ 66 ] | Average | ||||||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Size | FLOPs(G)↓ | ↓ | ↑ | ↑ | ↑ | ↑ | ↓ | ↑ | ↑ | ↑ | ↑ | ↓ | ↑ | ↑ | ↑ | ↑ | ↓ | ↑ | ↑ | ↑ | ↑ |
| (1, 1, 1, 1) | 118.4 | - | - | - | - | - | - | - | - | - | - | - | - | - | - | - | - | - | - | - | - |
| (4, 4, 4, 4) | 118.3 | .025 | .939 | .957 | .935 | .924 | .059 | .880 | .912 | .883 | .855 | .037 | .920 | .941 | .903 | .893 | .040 | .913 | .937 | .907 | .891 |
| (12, 12, 6, 6) | 118.3 | .023 | .941 | .967 | .932 | .931 | .048 | .891 | .928 | .890 | .873 | .032 | .931 | .952 | .910 | .911 | .034 | .921 | .949 | .911 | .905 |
| (12, 12, 12, 12) | 118.3 | .024 | .938 | .964 | .930 | .927 | .046 | .891 | .926 | .889 | .872 | .034 | .925 | .947 | .905 | .902 | .035 | .918 | .946 | .908 | .900 |
| (24, 24, 12, 12) | 118.3 | .023 | . 943 | .958 | .936 | .926 | .048 | .891 | .925 | .886 | .870 | .031 | .930 | .954 | .911 | .904 | .034 | .921 | .946 | .911 | .900 |
| NJUD [ 53 ] | LFSD [ 65 ] | SIP [ 66 ] | Average | |||||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Superpixel method | ↓ | ↑ | ↑ | ↑ | ↑ | ↓ | ↑ | ↑ | ↑ | ↑ | ↓ | ↑ | ↑ | ↑ | ↑ | ↓ | ↑ | ↑ | ↑ | ↑ |
| K-means clustering [ 52 ] | .025 | .938 | .962 | .928 | .925 | .050 | .891 | .926 | .883 | .868 | .035 | .927 | .947 | .903 | .903 | .037 | .919 | .945 | .905 | .899 |
| Self-attention based method [ 49 ] | .028 | .927 | .957 | .922 | .916 | .054 | .872 | .913 | .877 | .854 | .037 | .920 | .941 | .898 | .895 | .040 | .906 | .937 | .899 | .888 |
| Cross-attention based method [ 51 ] | .024 | .941 | .957 | .932 | .925 | .053 | .877 | .916 | .882 | .858 | .033 | .921 | .950 | .902 | .911 | .036 | .916 | .942 | .908 | .898 |
| Ours | .023 | .941 | .967 | .932 | .931 | .048 | .891 | .928 | .890 | .873 | .032 | .931 | .952 | .910 | .911 | .034 | .921 | .949 | .911 | .905 |
| NJUD [ 53 ] | LFSD [ 65 ] | SIP [ 66 ] | Average | |||||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Mask size | ↓ | ↑ | ↑ | ↑ | ↑ | ↓ | ↑ | ↑ | ↑ | ↑ | ↓ | ↑ | ↑ | ↑ | ↑ | ↓ | ↑ | ↑ | ↑ | ↑ |
| 3 3 | .024 | .940 | .965 | .931 | .926 | .048 | .889 | .923 | .888 | .872 | .034 | .926 | .948 | .903 | .901 | .035 | .918 | .945 | .907 | .900 |
| 5 5 | .023 | .941 | .967 | .932 | .931 | .048 | .891 | .928 | .890 | .873 | .032 | .931 | .952 | .910 | .911 | .034 | .921 | .949 | .911 | .905 |
| 7 7 | .029 | .923 | .955 | .919 | .911 | .053 | .883 | .921 | .879 | .862 | .037 | .916 | .942 | .900 | .893 | .040 | .907 | .939 | .899 | .889 |
Salient Object Detection
Salient Object Detection finds visually prominent objects in images or video and predicts their pixel-level foreground masks.
See all 37 papers