SYNTOLOGY HomeExplorerAtlasCodeMethodologyAboutDevelopersFeedPricing
Paper · 2406.17815 · 2024

SUM: Saliency Unification through Mamba for Visual Attention Modeling

arXiv · PDF · Open in the Atlas

Code that ran

We lifted 14 functions out of this paper's own repositories and ran 14 of them in a sandbox. "Ran" means the function executed on a synthesized input and returned a value. It is not a reproduction of the paper's results.

RepositoryRoleRan
Arhosseini77/SUM canonical 14 of 14
FunctionStatusWhere it lives
flops_selective_scan_ref Ran Arhosseini77/SUM/net/models/vmamba.py
code served (permissive licence) · get_code("c00dcf8428ee981b")
get_logger Ran Arhosseini77/SUM/net/utils.py
code served (permissive licence) · get_code("23a1f34627822bb3")
get_optimizer Ran Arhosseini77/SUM/net/utils.py
code served (permissive licence) · get_code("ae1a2dd9d62e7fad")
get_scheduler Ran Arhosseini77/SUM/net/utils.py
code served (permissive licence) · get_code("dea42821258184c2")
load_and_preprocess_image Ran Arhosseini77/SUM/net/inference_utils.py
code served (permissive licence) · get_code("a214ec961739db31")
loss_CC Ran Arhosseini77/SUM/utils/loss_function.py
code served (permissive licence) · get_code("774d792a6525842a")
loss_KLdiv Ran Arhosseini77/SUM/utils/loss_function.py
code served (permissive licence) · get_code("e9183c75893d49f0")
loss_similarity Ran Arhosseini77/SUM/utils/loss_function.py
code served (permissive licence) · get_code("9596d79dd2070fff")
mean_std Ran Arhosseini77/SUM/optuna_network.py
code served (permissive licence) · get_code("f0c787b1471d4421")
modulate Ran Arhosseini77/SUM/net/models/vmamba.py
code served (permissive licence) · get_code("95949c1af8f08e9d")
modulate_scale Ran Arhosseini77/SUM/net/models/vmamba.py
code served (permissive licence) · get_code("e26191359b0c82c3")
postprocess_img Ran Arhosseini77/SUM/utils/data_process_uni.py
code served (permissive licence) · get_code("ebfa76e7f72447ce")
predict_saliency_map Ran Arhosseini77/SUM/net/inference_utils.py
code served (permissive licence) · get_code("40e257a513360d15")
preprocess_img Ran Arhosseini77/SUM/utils/data_process_uni.py
code served (permissive licence) · get_code("f3af41b9e983a72c")

Repositories linked to this paper

Some links come from the archived Papers with Code dataset (CC BY-SA 4.0): attribution and licence.

Abstract

Visual attention modeling, important for interpreting and prioritizing visual stimuli, plays a significant role in applications such as marketing, multimedia, and robotics. Traditional saliency prediction models, especially those based on Convolutional Neural Networks (CNNs) or Transformers, achieve notable success by leveraging large-scale annotated datasets. However, the current state-of-the-art (SOTA) models that use Transformers are computationally expensive. Additionally, separate models are often required for each image type, lacking a unified approach. In this paper, we propose Saliency Unification through Mamba (SUM), a novel approach that integrates the efficient long-range dependency modeling of Mamba with U-Net to provide a unified model for diverse image types. Using a novel Conditional Visual State Space (C-VSS) block, SUM dynamically adapts to various image types, including natural scenes, web pages, and commercial imagery, ensuring universal applicability across different data types. Our comprehensive evaluations across five benchmarks demonstrate that SUM seamlessly adapts to different visual characteristics and consistently outperforms existing models. These results position SUM as a versatile and powerful tool for advancing visual attention modeling, offering a robust solution universally applicable across different types of visual content.

For agents

The same record, over MCP at https://syntology.ai/mcp:

get_harvested_code_for_paper("2406.17815")
get_code_for_paper("2406.17815")
have("2406.17815")

Connect an agent — have() is free.