{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/arxiv-2603-21879","title":"SmaAT-QMix-UNet: A Parameter-Efficient Vector-Quantized UNet for Precipitation Nowcasting","arxiv_id":"2603.21879","date":"2026-03-23","proceeding":null,"authors":["Nikolas Stavrou","Siamak Mehrkanoon"],"abstract":"Weather forecasting supports critical socioeconomic activities and complements environmental protection, yet operational Numerical Weather Prediction (NWP) systems remain computationally intensive, thus being inefficient for certain applications. Meanwhile, recent advances in deep data-driven models have demonstrated promising results in nowcasting tasks. This paper presents SmaAT-QMix-UNet, an enhanced variant of SmaAT-UNet that introduces two key innovations: a vector quantization (VQ) bottleneck at the encoder-decoder bridge, and mixed kernel depth-wise convolutions (MixConv) replacing selected encoder and decoder blocks. These enhancements both reduce the model's size and improve its nowcasting performance. We train and evaluate SmaAT-QMix-UNet on a Dutch radar precipitation dataset (2016-2019), predicting precipitation 30 minutes ahead. Three configurations are benchmarked: using only VQ, only MixConv, and the full SmaAT-QMix-UNet. Grad-CAM saliency maps highlight the regions influencing each nowcast, while a UMAP embedding of the codewords illustrates how the VQ layer clusters encoder outputs. The source code for SmaAT-QMix-UNet is publicly available on GitHub: https://github.com/nstavr04/MasterThesisSnellius.","url_abs":"https://arxiv.org/abs/2603.21879","url_pdf":"https://arxiv.org/pdf/2603.21879","source":{"archive":null,"snapshot":"2025-07-28","note":"not in the Papers with Code archive (frozen at the snapshot)","row_kind":"graph","title_abstract_authors_date":"arXiv metadata, CC0 1.0 (https://info.arxiv.org/help/license)"},"code_links":[],"tasks":[],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2603.21879","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2603.21879"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-24T18:15:14+00:00","read_at_is":"when the build read Syntology's graph, not when any sample ran","claim":"Per-sample execution status on synthesized fixtures; not a correctness claim about the paper. Samples come from repositories linked to the paper, official or community; repo_kind says which.","repos":[{"mentioned_in_github":null,"is_official":null,"provenance":"deterministic:regex_extraction","mentioned_in_paper":null,"url":"https://github.com/nstavr04/MasterThesisSnellius","reach":null}],"summary":{"ran":15,"ran_honours":1,"unverified":4},"by_repo_kind":{"found_in_text":{"samples":20,"ran":16,"repositories":1}},"repo_kind_vocabulary":{"official":"The archive marks this repository official for the paper","named_in_paper":"The archive records that the paper mentions this repository; it is not marked official","listed":"In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper","found_in_text":"Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted","community":"Not in the archive's code links for this paper; a community repository Syntology harvested"},"n_pointer_only_for_licence":20,"samples":[{"code_sha256_prefix":"f8a29047ca79e69c","entry":"CBAM","repo":"nstavr04/MasterThesisSnellius","repo_kind":"found_in_text","path":"SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","file_url":"https://github.com/nstavr04/MasterThesisSnellius/blob/HEAD/SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"f8a29047ca79e69c"}},{"code_sha256_prefix":"9b35cac2322cc879","entry":"ChannelAttention","repo":"nstavr04/MasterThesisSnellius","repo_kind":"found_in_text","path":"SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","file_url":"https://github.com/nstavr04/MasterThesisSnellius/blob/HEAD/SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"9b35cac2322cc879"}},{"code_sha256_prefix":"15b2d26784c296c6","entry":"DepthwiseSeparableConv","repo":"nstavr04/MasterThesisSnellius","repo_kind":"found_in_text","path":"SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","file_url":"https://github.com/nstavr04/MasterThesisSnellius/blob/HEAD/SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"15b2d26784c296c6"}},{"code_sha256_prefix":"8452e3e27b98a2b2","entry":"DoubleConvDS","repo":"nstavr04/MasterThesisSnellius","repo_kind":"found_in_text","path":"SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","file_url":"https://github.com/nstavr04/MasterThesisSnellius/blob/HEAD/SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"8452e3e27b98a2b2"}},{"code_sha256_prefix":"0979d140721d5051","entry":"DoubleMixConv","repo":"nstavr04/MasterThesisSnellius","repo_kind":"found_in_text","path":"SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","file_url":"https://github.com/nstavr04/MasterThesisSnellius/blob/HEAD/SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"0979d140721d5051"}},{"code_sha256_prefix":"8c8ba3857fccc505","entry":"DownDS","repo":"nstavr04/MasterThesisSnellius","repo_kind":"found_in_text","path":"SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","file_url":"https://github.com/nstavr04/MasterThesisSnellius/blob/HEAD/SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"8c8ba3857fccc505"}},{"code_sha256_prefix":"3bbe039ca4787ea1","entry":"MDConv","repo":"nstavr04/MasterThesisSnellius","repo_kind":"found_in_text","path":"SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","file_url":"https://github.com/nstavr04/MasterThesisSnellius/blob/HEAD/SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"3bbe039ca4787ea1"}},{"code_sha256_prefix":"83215e990e67b6b7","entry":"MixDownDS","repo":"nstavr04/MasterThesisSnellius","repo_kind":"found_in_text","path":"SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","file_url":"https://github.com/nstavr04/MasterThesisSnellius/blob/HEAD/SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"83215e990e67b6b7"}},{"code_sha256_prefix":"f7eeb12c5ef3eafa","entry":"MixUpDS","repo":"nstavr04/MasterThesisSnellius","repo_kind":"found_in_text","path":"SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","file_url":"https://github.com/nstavr04/MasterThesisSnellius/blob/HEAD/SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"f7eeb12c5ef3eafa"}},{"code_sha256_prefix":"09f3a772fe136cd3","entry":"OutConv","repo":"nstavr04/MasterThesisSnellius","repo_kind":"found_in_text","path":"SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","file_url":"https://github.com/nstavr04/MasterThesisSnellius/blob/HEAD/SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"09f3a772fe136cd3"}},{"code_sha256_prefix":"7af9b82594a1f7c4","entry":"SpatialAttention","repo":"nstavr04/MasterThesisSnellius","repo_kind":"found_in_text","path":"SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","file_url":"https://github.com/nstavr04/MasterThesisSnellius/blob/HEAD/SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"invariant","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"7af9b82594a1f7c4"}},{"code_sha256_prefix":"04721935113ff9af","entry":"UpDS","repo":"nstavr04/MasterThesisSnellius","repo_kind":"found_in_text","path":"SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","file_url":"https://github.com/nstavr04/MasterThesisSnellius/blob/HEAD/SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"04721935113ff9af"}},{"code_sha256_prefix":"536271aae35f2e3b","entry":"VectorQuantizer","repo":"nstavr04/MasterThesisSnellius","repo_kind":"found_in_text","path":"SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","file_url":"https://github.com/nstavr04/MasterThesisSnellius/blob/HEAD/SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"536271aae35f2e3b"}},{"code_sha256_prefix":"701db614781db3dd","entry":"precipitation_maps_h5","repo":"nstavr04/MasterThesisSnellius","repo_kind":"found_in_text","path":"SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","file_url":"https://github.com/nstavr04/MasterThesisSnellius/blob/HEAD/SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"701db614781db3dd"}},{"code_sha256_prefix":"17d1720b12b2160b","entry":"precipitation_maps_oversampled_h5","repo":"nstavr04/MasterThesisSnellius","repo_kind":"found_in_text","path":"SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","file_url":"https://github.com/nstavr04/MasterThesisSnellius/blob/HEAD/SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"17d1720b12b2160b"}},{"code_sha256_prefix":"f22062eccb4ebd7f","entry":"split_layer","repo":"nstavr04/MasterThesisSnellius","repo_kind":"found_in_text","path":"SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","file_url":"https://github.com/nstavr04/MasterThesisSnellius/blob/HEAD/SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","link_basis":"first_harvest_node","language":"python","status":"ran_honours","verification_level":1,"contract_check":"HONOURS","metamorphic_tier":"well_formed","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"f22062eccb4ebd7f"}},{"code_sha256_prefix":"02305a1d2b842e23","entry":"DepthwiseConv2D","repo":"nstavr04/MasterThesisSnellius","repo_kind":"found_in_text","path":"SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","file_url":"https://github.com/nstavr04/MasterThesisSnellius/blob/HEAD/SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"02305a1d2b842e23"}},{"code_sha256_prefix":"00f2156eea1119a1","entry":"Precip_regression_base","repo":"nstavr04/MasterThesisSnellius","repo_kind":"found_in_text","path":"SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","file_url":"https://github.com/nstavr04/MasterThesisSnellius/blob/HEAD/SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"00f2156eea1119a1"}},{"code_sha256_prefix":"f0459421070e43fd","entry":"SmaAT_UNet_MixConv_VQ","repo":"nstavr04/MasterThesisSnellius","repo_kind":"found_in_text","path":"SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","file_url":"https://github.com/nstavr04/MasterThesisSnellius/blob/HEAD/SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"f0459421070e43fd"}},{"code_sha256_prefix":"1fad264603540926","entry":"UNet_base","repo":"nstavr04/MasterThesisSnellius","repo_kind":"found_in_text","path":"SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","file_url":"https://github.com/nstavr04/MasterThesisSnellius/blob/HEAD/SmaAT-QMix-UNet/models/SmaAT_UNet_VQ_lightning.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"1fad264603540926"}}]},"arxiv_metadata":{"licence":"arXiv metadata, CC0 1.0 (https://info.arxiv.org/help/license)","fields":["title","abstract","authors","date"],"primary_category":"cs.LG","source":"arxiv_2026.jsonl"},"syntology_extracted_results":null}