Super-squash branch 'main' using huggingface_hub

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,18 @@
+# SPBench Model
+
+Pre-trained model weights for seismic processing benchmarks.
+
+Each subdirectory contains a complete model repository (weights, configs, and the original model card as `README.md`):
+
+| Subdirectory | Task |
+|---|---|
+| [ground-roll-attenuation](ground-roll-attenuation) | Ground roll attenuation |
+| [seismic-interpolation](seismic-interpolation) | Seismic trace interpolation |
+| [random-noise-attenuation](random-noise-attenuation) | Random noise attenuation |
+| [first-break-picking](first-break-picking) | First break picking |
+| [multiples-attenuation](multiples-attenuation) | Multiples attenuation |
+| [blending-noise-suppression](blending-noise-suppression) | Blending noise suppression |
+| [blending-noise-suppression-avo](blending-noise-suppression-avo) | Blending noise suppression (AVO) |
+| [random-noise-attenuation-avo](random-noise-attenuation-avo) | Random noise attenuation (AVO) |
+| [ground-roll-attenuation-field](ground-roll-attenuation-field) | Ground roll attenuation (field data) |
+| [seismic-interpolation-benchmark](seismic-interpolation-benchmark) | Seismic interpolation benchmark |

blending-noise-suppression-avo/.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

blending-noise-suppression-avo/README.md

@@ -0,0 +1,1667 @@
+---
+tags:
+- seismic
+- blending-noise
+- deblending
+- avo
+- common-receiver
+- unet
+- resunet
+- dncnn
+- attention-unet
+- pytorch
+library_name: pytorch
+---
+
+# AVO Blending Noise Suppression Benchmark
+
+This task performs supervised deblending on paired AVO common-receiver gathers.
+
+## Task
+
+The model directly learns a mapping from pseudo-deblended input to the clean reference:
+
+```python
+denoised = model(pseudo_deblended_input)
+```
+
+This is a paired regression task, not a synthetic noise-label regression task.
+
+## Dataset
+
+Current paired SEG-Y inputs:
+
+- input: `/data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy`
+- target: `/data/data/T03_seismic_common_receiver.sgy`
+
+Logical volume shape:
+
+- `(120, 1001, 1500)`
+  - 120 common-receiver gathers
+  - 1001 traces per common-receiver gather
+  - 1500 time samples per trace
+
+Split policy:
+
+- sequential receiver-gather split
+- train: 96
+- val: 12
+- test: 12
+
+## Models
+
+- `unet`
+- `res_unet`
+- `dncnn`
+- `atten_unet`
+- `ddpm`
+- `unet_L`
+- `SCRN`
+
+All model definitions are loaded from `model/blending_noise_suppression/`.
+
+## Preprocessing
+
+- paired loading with shape consistency check
+- shared normalization statistics for input and target
+- `max_abs` normalization
+- per-gather normalization through the shared `shot` normalization scope
+- overlapping 2D patches of size `128 x 256`
+- overlap ratio `0.5`
+- FB-FRE uses `dt: 0.004` and adaptive tapering via `taper_width: null`
+
+## Scripts
+
+Training:
+
+- `bash scripts/blending_noise_suppression_avo/train_denoise_unet.sh`
+- `bash scripts/blending_noise_suppression_avo/train_denoise_res_unet.sh`
+- `bash scripts/blending_noise_suppression_avo/train_denoise_dncnn.sh`
+- `bash scripts/blending_noise_suppression_avo/train_denoise_atten_unet.sh`
+- `bash scripts/blending_noise_suppression_avo/train_denoise_ddpm.sh`
+- `bash scripts/blending_noise_suppression_avo/train_denoise_unet_L.sh`
+- `bash scripts/blending_noise_suppression_avo/train_denoise_SCRN.sh`
+
+Each training script runs a fixed three-seed sweep:
+
+- seeds: `42`, `43`, `44`
+- run names: `<experiment.name>_T03_avo_mod_seed<seed>_T03_pseudo_deblended_seismic_common_receiver_mod`
+- checkpoints: `/data/data/results/blending_noise_avo/<model>/<run_name>/checkpoints/`
+
+Inference:
+
+- `bash scripts/blending_noise_suppression_avo/inference_denoise_unet.sh`
+- `bash scripts/blending_noise_suppression_avo/inference_denoise_res_unet.sh`
+- `bash scripts/blending_noise_suppression_avo/inference_denoise_dncnn.sh`
+- `bash scripts/blending_noise_suppression_avo/inference_denoise_atten_unet.sh`
+- `bash scripts/blending_noise_suppression_avo/inference_denoise_ddpm.sh`
+- `bash scripts/blending_noise_suppression_avo/inference_denoise_unet_L.sh`
+- `bash scripts/blending_noise_suppression_avo/inference_denoise_SCRN.sh`
+
+Each inference script matches the same three seeds, writes per-seed outputs to:
+
+- `/data/data/results/blending_noise_avo/<model>/<run_name>/inference_integrated_metrics`
+
+and also saves a cross-seed aggregate summary to:
+
+- `/data/data/results/blending_noise_avo/<model>/<experiment.name>_T03_avo_mod_seed_stats/metrics_summary_mean_std.json`
+
+Run all:
+
+- `bash scripts/blending_noise_suppression_avo/run_all_blending_models.sh`
+
+## Upload
+
+Dataset upload helper:
+
+- `bash scripts/blending_noise_suppression_avo/upload_dataset_to_hf.sh`
+
+This uploads:
+
+- `input/T03_pseudo_deblended_seismic_common_receiver_mod.sgy`
+- `target/T03_seismic_common_receiver.sgy`
+- generated dataset card `README.md`
+
+Model upload helper:
+
+- `bash scripts/blending_noise_suppression_avo/upload_blending_model_to_hf.sh`
+
+This uploads:
+
+- `models/<model>/T03_avo_mod_seed<seed>/best.pt`
+- `models/<model>/T03_avo_mod_seed<seed>/config.yaml`
+- generated model card `README.md`
+
+## Outputs
+
+Each inference directory contains:
+
+- `metrics_per_gather.csv`
+- `metrics_per_sample.csv`
+- `metrics_summary.json`
+- `visualizations/`
+- optional `npy/`
+
+Metrics are reported for:
+
+- input vs target
+- denoised vs target
+- delta = denoised minus input
+
+For naming consistency:
+
+- `metrics_summary.json` uses `input`, `denoised`, and `delta`
+- the file also keeps a backward-compatible `restored` alias
+- `metrics_per_sample.csv` provides a generic sample-level view, while `metrics_per_gather.csv` keeps the gather-specific naming
+
+<!-- AUTO_RESULTS_BEGIN -->
+## Results
+
+Mean +- std over available seeds, computed from `*_seed_stats/metrics_summary_mean_std.json`.
+Metrics are reported on common-receiver gathers in the normalized domain.
+
+### Dataset Variant `T03_avo_mod`
+
+| Method | Parameters (M) | SNR | PSNR | SSIM | MAE | MSE | RMSE | EB_WSE_MEDIUM_40_70_NE | EB_WSE_MEDIUM_40_70_SNR | EB_WSE_STRONG_70_100_NE | EB_WSE_STRONG_70_100_SNR | EB_WSE_VERY_WEAK_5_20_NE | EB_WSE_VERY_WEAK_5_20_SNR | EB_WSE_WEAK_20_40_NE | EB_WSE_WEAK_20_40_SNR | FB_FRE_HIGH_ENERGY_RATIO | FB_FRE_HIGH_FREQUENCY_RANGE_HZ | FB_FRE_HIGH_NE | FB_FRE_HIGH_SNR | FB_FRE_LOW_ENERGY_RATIO | FB_FRE_LOW_FREQUENCY_RANGE_HZ | FB_FRE_LOW_NE | FB_FRE_LOW_SNR | FB_FRE_MID_ENERGY_RATIO | FB_FRE_MID_FREQUENCY_RANGE_HZ | FB_FRE_MID_NE | FB_FRE_MID_SNR | FB_FRE_VERY_HIGH_ENERGY_RATIO | FB_FRE_VERY_HIGH_FREQUENCY_RANGE_HZ | FB_FRE_VERY_HIGH_NE | FB_FRE_VERY_HIGH_SNR |
+|---|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|
+| Input | - | -2.0292+-0.0000 | 23.1464+-0.0000 | 0.7028+-0.0000 | 0.025431+-0.000000 | 0.005029+-0.000000 | 0.070259+-0.000000 | 16.207542+-0.000000 | -24.1817+-0.0000 | 0.211915+-0.000000 | 13.5116+-0.0000 | 131.715259+-0.000000 | -42.3526+-0.0000 | 54.756158+-0.000000 | -34.7438+-0.0000 | 0.140025+-0.000000 | 48.25-73.3 | 1.263001+-0.000000 | -2.0278+-0.0000 | 0.125627+-0.000000 | 6.5-23.2 | 1.251054+-0.000000 | -1.9452+-0.0000 | 0.637894+-0.000000 | 23.2-48.25 | 1.265097+-0.000000 | -2.0422+-0.0000 | 0.001545+-0.000000 | 73.3-90 | 1.300019+-0.000000 | -2.2787+-0.0000 |
+| UNet | 7.76 | 19.8927+-0.6858 | 45.0683+-0.6858 | 0.9886+-0.0012 | 0.003580+-0.000311 | 0.000033+-0.000005 | 0.005644+-0.000456 | 0.924552+-0.029280 | 0.6885+-0.2748 | 0.081971+-0.006676 | 21.7567+-0.6961 | 4.448344+-0.984280 | -12.7468+-1.8217 | 2.109300+-0.386105 | -6.3407+-1.5137 | 0.140025+-0.000000 | 48.25-73.3 | 0.110440+-0.012523 | 19.2170+-0.9686 | 0.125627+-0.000000 | 6.5-23.2 | 0.128963+-0.003627 | 17.7975+-0.2436 | 0.637894+-0.000000 | 23.2-48.25 | 0.074045+-0.006893 | 22.6466+-0.7862 | 0.001545+-0.000000 | 73.3-90 | 0.672353+-0.045598 | 3.5418+-0.5725 |
+| DnCNN | 0.56 | 21.1427+-0.0832 | 46.3184+-0.0832 | 0.9879+-0.0003 | 0.003212+-0.000035 | 0.000024+-0.000000 | 0.004882+-0.000046 | 1.041794+-0.009072 | -0.3360+-0.0752 | 0.046714+-0.000249 | 26.6125+-0.0459 | 6.884248+-0.115049 | -16.6983+-0.1451 | 3.056675+-0.050152 | -9.6611+-0.1447 | 0.140025+-0.000000 | 48.25-73.3 | 0.084323+-0.001455 | 21.4974+-0.1497 | 0.125627+-0.000000 | 6.5-23.2 | 0.123886+-0.001142 | 18.1439+-0.0805 | 0.637894+-0.000000 | 23.2-48.25 | 0.067761+-0.000643 | 23.3821+-0.0822 | 0.001545+-0.000000 | 73.3-90 | 0.478678+-0.009963 | 6.4558+-0.1820 |
+| ResUNet | 8.11 | 19.1792+-0.4159 | 44.3549+-0.4159 | 0.9863+-0.0006 | 0.004009+-0.000166 | 0.000038+-0.000004 | 0.006109+-0.000288 | 1.172883+-0.031293 | -1.3759+-0.2336 | 0.073572+-0.004656 | 22.6840+-0.5632 | 7.385174+-0.356539 | -17.3254+-0.4249 | 3.264928+-0.163859 | -10.2494+-0.4437 | 0.140025+-0.000000 | 48.25-73.3 | 0.116582+-0.008488 | 18.7334+-0.6548 | 0.125627+-0.000000 | 6.5-23.2 | 0.137702+-0.003074 | 17.2280+-0.1959 | 0.637894+-0.000000 | 23.2-48.25 | 0.073755+-0.002954 | 22.6513+-0.3527 | 0.001545+-0.000000 | 73.3-90 | 0.796460+-0.049241 | 2.0905+-0.5491 |
+| Attention UNet | 7.85 | 20.5846+-0.4090 | 45.7603+-0.4090 | 0.9900+-0.0004 | 0.003216+-0.000116 | 0.000028+-0.000003 | 0.005203+-0.000255 | 0.851327+-0.007454 | 1.3988+-0.0763 | 0.077606+-0.004972 | 22.2240+-0.5415 | 3.095065+-0.207962 | -9.7852+-0.5872 | 1.551991+-0.078837 | -3.8028+-0.4482 | 0.140025+-0.000000 | 48.25-73.3 | 0.098203+-0.006586 | 20.1950+-0.5752 | 0.125627+-0.000000 | 6.5-23.2 | 0.122955+-0.002681 | 18.2104+-0.1877 | 0.637894+-0.000000 | 23.2-48.25 | 0.070269+-0.003938 | 23.0786+-0.4780 | 0.001545+-0.000000 | 73.3-90 | 0.584715+-0.024681 | 4.7398+-0.3712 |
+| DDPM | 33.30 | 17.6126+-0.6137 | 42.7883+-0.6137 | 0.9794+-0.0022 | 0.005497+-0.000301 | 0.000054+-0.000008 | 0.007285+-0.000513 | 1.385622+-0.080286 | -2.8122+-0.4963 | 0.073466+-0.003703 | 22.7173+-0.4265 | 11.004830+-1.232630 | -20.7351+-0.9303 | 4.621439+-0.447510 | -13.2309+-0.8139 | 0.140025+-0.000000 | 48.25-73.3 | 0.141598+-0.011702 | 17.1858+-0.6818 | 0.125627+-0.000000 | 6.5-23.2 | 0.155503+-0.008703 | 16.1821+-0.4784 | 0.637894+-0.000000 | 23.2-48.25 | 0.081625+-0.004615 | 21.7908+-0.4798 | 0.001545+-0.000000 | 73.3-90 | 1.070203+-0.095554 | -0.2780+-0.7348 |
+| UNet-L | 31.04 | 23.9298+-0.1370 | 49.1054+-0.1370 | 0.9929+-0.0004 | 0.002287+-0.000029 | 0.000013+-0.000000 | 0.003525+-0.000054 | 0.750301+-0.021546 | 2.5001+-0.2469 | 0.045716+-0.000730 | 26.8004+-0.1390 | 2.004984+-0.068431 | -6.0135+-0.2978 | 1.226751+-0.019845 | -1.7716+-0.1428 | 0.140025+-0.000000 | 48.25-73.3 | 0.052170+-0.001292 | 25.6964+-0.2126 | 0.125627+-0.000000 | 6.5-23.2 | 0.100387+-0.002986 | 19.9710+-0.2565 | 0.637894+-0.000000 | 23.2-48.25 | 0.051987+-0.000848 | 25.6835+-0.1410 | 0.001545+-0.000000 | 73.3-90 | 0.267608+-0.013954 | 11.5575+-0.4505 |
+| SCRN | 0.43 | 25.4852+-0.0375 | 50.6608+-0.0374 | 0.9953+-0.0000 | 0.001947+-0.000010 | 0.000009+-0.000000 | 0.002941+-0.000013 | 0.604353+-0.002911 | 4.3745+-0.0421 | 0.038012+-0.000002 | 28.4064+-0.0007 | 2.115244+-0.084896 | -6.4810+-0.3500 | 1.235941+-0.018991 | -1.8354+-0.1332 | 0.140025+-0.000000 | 48.25-73.3 | 0.044643+-0.000415 | 27.0526+-0.0863 | 0.125627+-0.000000 | 6.5-23.2 | 0.081920+-0.000374 | 21.7352+-0.0398 | 0.637894+-0.000000 | 23.2-48.25 | 0.046667+-0.000178 | 26.6210+-0.0334 | 0.001545+-0.000000 | 73.3-90 | 0.166604+-0.000429 | 15.6849+-0.0277 |
+| FFCNN | 0.56 | 10.0554+-0.0000 | 35.2311+-0.0000 | 0.9670+-0.0000 | 0.007205+-0.000000 | 0.000314+-0.000000 | 0.017515+-0.000000 | 1.026906+-0.000000 | -0.2234+-0.0000 | 0.308648+-0.000000 | 10.2140+-0.0000 | 3.212355+-0.000000 | -10.1196+-0.0000 | 1.800039+-0.000000 | -5.0900+-0.0000 | 0.140025+-0.000000 | 48.25-73.3 | 0.477545+-0.000000 | 6.4290+-0.0000 | 0.125627+-0.000000 | 6.5-23.2 | 0.262586+-0.000000 | 11.6211+-0.0000 | 0.637894+-0.000000 | 23.2-48.25 | 0.245133+-0.000000 | 12.2129+-0.0000 | 0.001545+-0.000000 | 73.3-90 | 1.570027+-0.000000 | -3.8827+-0.0000 |
+| FBResNet | 0.45 | 20.8769+-0.0000 | 46.0525+-0.0000 | 0.9882+-0.0000 | 0.003269+-0.000000 | 0.000025+-0.000000 | 0.005014+-0.000000 | 1.040688+-0.000000 | -0.3442+-0.0000 | 0.052394+-0.000000 | 25.6146+-0.0000 | 6.730613+-0.000000 | -16.5438+-0.0000 | 3.019342+-0.000000 | -9.5839+-0.0000 | 0.140025+-0.000000 | 48.25-73.3 | 0.090167+-0.000000 | 20.9376+-0.0000 | 0.125627+-0.000000 | 6.5-23.2 | 0.122403+-0.000000 | 18.2453+-0.0000 | 0.637894+-0.000000 | 23.2-48.25 | 0.070533+-0.000000 | 23.0324+-0.0000 | 0.001545+-0.000000 | 73.3-90 | 0.502504+-0.000000 | 6.0915+-0.0000 |
+| q_unet | 21.88 | 21.7835+-0.4410 | 46.9591+-0.4410 | 0.9916+-0.0002 | 0.002760+-0.000075 | 0.000021+-0.000002 | 0.004533+-0.000236 | 0.806676+-0.008600 | 1.8678+-0.0914 | 0.066082+-0.005477 | 23.6310+-0.7017 | 2.228961+-0.257300 | -6.8951+-1.0203 | 1.284304+-0.077039 | -2.1563+-0.5275 | 0.140025+-0.000000 | 48.25-73.3 | 0.079658+-0.006209 | 22.0223+-0.6632 | 0.125627+-0.000000 | 6.5-23.2 | 0.113970+-0.003321 | 18.8705+-0.2531 | 0.637894+-0.000000 | 23.2-48.25 | 0.061408+-0.001700 | 24.2392+-0.2383 | 0.001545+-0.000000 | 73.3-90 | 0.500757+-0.053148 | 6.1082+-0.8977 |
+| unet_plusplus | 9.05 | 24.1199+-0.2423 | 49.2956+-0.2423 | 0.9923+-0.0004 | 0.002253+-0.000057 | 0.000012+-0.000001 | 0.003447+-0.000097 | 0.757421+-0.028589 | 2.4215+-0.3287 | 0.042676+-0.001080 | 27.4034+-0.2230 | 2.291199+-0.019998 | -7.1792+-0.0743 | 1.344684+-0.007086 | -2.5700+-0.0451 | 0.140025+-0.000000 | 48.25-73.3 | 0.046849+-0.001166 | 26.6284+-0.2227 | 0.125627+-0.000000 | 6.5-23.2 | 0.104241+-0.002996 | 19.6444+-0.2517 | 0.637894+-0.000000 | 23.2-48.25 | 0.051590+-0.001425 | 25.7523+-0.2407 | 0.001545+-0.000000 | 73.3-90 | 0.213581+-0.005819 | 13.5175+-0.2481 |
+
+<!-- AUTO_RESULTS_DATA_BEGIN
+{
+  "groups": {
+    "T03_avo_mod": {
+      "input": {
+        "eb_wse_medium_40_70_ne": {
+          "mean": 16.207542,
+          "std": 0.0
+        },
+        "eb_wse_medium_40_70_snr": {
+          "mean": -24.181704,
+          "std": 0.0
+        },
+        "eb_wse_strong_70_100_ne": {
+          "mean": 0.211915,
+          "std": 0.0
+        },
+        "eb_wse_strong_70_100_snr": {
+          "mean": 13.511556,
+          "std": 0.0
+        },
+        "eb_wse_very_weak_5_20_ne": {
+          "mean": 131.715259,
+          "std": 0.0
+        },
+        "eb_wse_very_weak_5_20_snr": {
+          "mean": -42.35256,
+          "std": 0.0
+        },
+        "eb_wse_weak_20_40_ne": {
+          "mean": 54.756158,
+          "std": 0.0
+        },
+        "eb_wse_weak_20_40_snr": {
+          "mean": -34.743819,
+          "std": 0.0
+        },
+        "fb_fre_high_energy_ratio": {
+          "mean": 0.140025,
+          "std": 0.0
+        },
+        "fb_fre_high_frequency_range_hz": [
+          48.25,
+          73.3
+        ],
+        "fb_fre_high_ne": {
+          "mean": 1.263001,
+          "std": 0.0
+        },
+        "fb_fre_high_snr": {
+          "mean": -2.027766,
+          "std": 0.0
+        },
+        "fb_fre_low_energy_ratio": {
+          "mean": 0.125627,
+          "std": 0.0
+        },
+        "fb_fre_low_frequency_range_hz": [
+          6.5,
+          23.2
+        ],
+        "fb_fre_low_ne": {
+          "mean": 1.251054,
+          "std": 0.0
+        },
+        "fb_fre_low_snr": {
+          "mean": -1.945173,
+          "std": 0.0
+        },
+        "fb_fre_mid_energy_ratio": {
+          "mean": 0.637894,
+          "std": 0.0
+        },
+        "fb_fre_mid_frequency_range_hz": [
+          23.2,
+          48.25
+        ],
+        "fb_fre_mid_ne": {
+          "mean": 1.265097,
+          "std": 0.0
+        },
+        "fb_fre_mid_snr": {
+          "mean": -2.042182,
+          "std": 0.0
+        },
+        "fb_fre_very_high_energy_ratio": {
+          "mean": 0.001545,
+          "std": 0.0
+        },
+        "fb_fre_very_high_frequency_range_hz": [
+          73.3,
+          90.0
+        ],
+        "fb_fre_very_high_ne": {
+          "mean": 1.300019,
+          "std": 0.0
+        },
+        "fb_fre_very_high_snr": {
+          "mean": -2.278716,
+          "std": 0.0
+        },
+        "mae": {
+          "mean": 0.025431,
+          "std": 0.0
+        },
+        "mse": {
+          "mean": 0.005029,
+          "std": 0.0
+        },
+        "psnr": {
+          "mean": 23.1464,
+          "std": 0.0
+        },
+        "rmse": {
+          "mean": 0.070259,
+          "std": 0.0
+        },
+        "snr": {
+          "mean": -2.0292,
+          "std": 0.0
+        },
+        "ssim": {
+          "mean": 0.7028,
+          "std": 0.0
+        }
+      },
+      "models": {
+        "unet": {
+          "eb_wse_medium_40_70_ne": {
+            "mean": 0.924552,
+            "std": 0.02928
+          },
+          "eb_wse_medium_40_70_snr": {
+            "mean": 0.688472,
+            "std": 0.274839
+          },
+          "eb_wse_strong_70_100_ne": {
+            "mean": 0.081971,
+            "std": 0.006676
+          },
+          "eb_wse_strong_70_100_snr": {
+            "mean": 21.756696,
+            "std": 0.696143
+          },
+          "eb_wse_very_weak_5_20_ne": {
+            "mean": 4.448344,
+            "std": 0.98428
+          },
+          "eb_wse_very_weak_5_20_snr": {
+            "mean": -12.74683,
+            "std": 1.821724
+          },
+          "eb_wse_weak_20_40_ne": {
+            "mean": 2.1093,
+            "std": 0.386105
+          },
+          "eb_wse_weak_20_40_snr": {
+            "mean": -6.340672,
+            "std": 1.513677
+          },
+          "fb_fre_high_energy_ratio": {
+            "mean": 0.140025,
+            "std": 0.0
+          },
+          "fb_fre_high_frequency_range_hz": [
+            48.25,
+            73.3
+          ],
+          "fb_fre_high_ne": {
+            "mean": 0.11044,
+            "std": 0.012523
+          },
+          "fb_fre_high_snr": {
+            "mean": 19.217039,
+            "std": 0.96862
+          },
+          "fb_fre_low_energy_ratio": {
+            "mean": 0.125627,
+            "std": 0.0
+          },
+          "fb_fre_low_frequency_range_hz": [
+            6.5,
+            23.2
+          ],
+          "fb_fre_low_ne": {
+            "mean": 0.128963,
+            "std": 0.003627
+          },
+          "fb_fre_low_snr": {
+            "mean": 17.797461,
+            "std": 0.243554
+          },
+          "fb_fre_mid_energy_ratio": {
+            "mean": 0.637894,
+            "std": 0.0
+          },
+          "fb_fre_mid_frequency_range_hz": [
+            23.2,
+            48.25
+          ],
+          "fb_fre_mid_ne": {
+            "mean": 0.074045,
+            "std": 0.006893
+          },
+          "fb_fre_mid_snr": {
+            "mean": 22.64655,
+            "std": 0.786202
+          },
+          "fb_fre_very_high_energy_ratio": {
+            "mean": 0.001545,
+            "std": 0.0
+          },
+          "fb_fre_very_high_frequency_range_hz": [
+            73.3,
+            90.0
+          ],
+          "fb_fre_very_high_ne": {
+            "mean": 0.672353,
+            "std": 0.045598
+          },
+          "fb_fre_very_high_snr": {
+            "mean": 3.541757,
+            "std": 0.572457
+          },
+          "mae": {
+            "mean": 0.00358,
+            "std": 0.000311
+          },
+          "mse": {
+            "mean": 3.3e-05,
+            "std": 5e-06
+          },
+          "psnr": {
+            "mean": 45.068333,
+            "std": 0.68579
+          },
+          "rmse": {
+            "mean": 0.005644,
+            "std": 0.000456
+          },
+          "snr": {
+            "mean": 19.892667,
+            "std": 0.685806
+          },
+          "ssim": {
+            "mean": 0.988633,
+            "std": 0.001156
+          }
+        },
+        "dncnn": {
+          "eb_wse_medium_40_70_ne": {
+            "mean": 1.041794,
+            "std": 0.009072
+          },
+          "eb_wse_medium_40_70_snr": {
+            "mean": -0.335971,
+            "std": 0.075225
+          },
+          "eb_wse_strong_70_100_ne": {
+            "mean": 0.046714,
+            "std": 0.000249
+          },
+          "eb_wse_strong_70_100_snr": {
+            "mean": 26.612542,
+            "std": 0.045882
+          },
+          "eb_wse_very_weak_5_20_ne": {
+            "mean": 6.884248,
+            "std": 0.115049
+          },
+          "eb_wse_very_weak_5_20_snr": {
+            "mean": -16.698266,
+            "std": 0.145059
+          },
+          "eb_wse_weak_20_40_ne": {
+            "mean": 3.056675,
+            "std": 0.050152
+          },
+          "eb_wse_weak_20_40_snr": {
+            "mean": -9.661108,
+            "std": 0.14471
+          },
+          "fb_fre_high_energy_ratio": {
+            "mean": 0.140025,
+            "std": 0.0
+          },
+          "fb_fre_high_frequency_range_hz": [
+            48.25,
+            73.3
+          ],
+          "fb_fre_high_ne": {
+            "mean": 0.084323,
+            "std": 0.001455
+          },
+          "fb_fre_high_snr": {
+            "mean": 21.497399,
+            "std": 0.149676
+          },
+          "fb_fre_low_energy_ratio": {
+            "mean": 0.125627,
+            "std": 0.0
+          },
+          "fb_fre_low_frequency_range_hz": [
+            6.5,
+            23.2
+          ],
+          "fb_fre_low_ne": {
+            "mean": 0.123886,
+            "std": 0.001142
+          },
+          "fb_fre_low_snr": {
+            "mean": 18.143944,
+            "std": 0.080544
+          },
+          "fb_fre_mid_energy_ratio": {
+            "mean": 0.637894,
+            "std": 0.0
+          },
+          "fb_fre_mid_frequency_range_hz": [
+            23.2,
+            48.25
+          ],
+          "fb_fre_mid_ne": {
+            "mean": 0.067761,
+            "std": 0.000643
+          },
+          "fb_fre_mid_snr": {
+            "mean": 23.382114,
+            "std": 0.082215
+          },
+          "fb_fre_very_high_energy_ratio": {
+            "mean": 0.001545,
+            "std": 0.0
+          },
+          "fb_fre_very_high_frequency_range_hz": [
+            73.3,
+            90.0
+          ],
+          "fb_fre_very_high_ne": {
+            "mean": 0.478678,
+            "std": 0.009963
+          },
+          "fb_fre_very_high_snr": {
+            "mean": 6.45578,
+            "std": 0.182005
+          },
+          "mae": {
+            "mean": 0.003212,
+            "std": 3.5e-05
+          },
+          "mse": {
+            "mean": 2.4e-05,
+            "std": 0.0
+          },
+          "psnr": {
+            "mean": 46.318367,
+            "std": 0.083205
+          },
+          "rmse": {
+            "mean": 0.004882,
+            "std": 4.6e-05
+          },
+          "snr": {
+            "mean": 21.1427,
+            "std": 0.08324
+          },
+          "ssim": {
+            "mean": 0.987933,
+            "std": 0.000262
+          }
+        },
+        "res_unet": {
+          "eb_wse_medium_40_70_ne": {
+            "mean": 1.172883,
+            "std": 0.031293
+          },
+          "eb_wse_medium_40_70_snr": {
+            "mean": -1.375916,
+            "std": 0.23355
+          },
+          "eb_wse_strong_70_100_ne": {
+            "mean": 0.073572,
+            "std": 0.004656
+          },
+          "eb_wse_strong_70_100_snr": {
+            "mean": 22.68396,
+            "std": 0.563248
+          },
+          "eb_wse_very_weak_5_20_ne": {
+            "mean": 7.385174,
+            "std": 0.356539
+          },
+          "eb_wse_very_weak_5_20_snr": {
+            "mean": -17.32537,
+            "std": 0.424895
+          },
+          "eb_wse_weak_20_40_ne": {
+            "mean": 3.264928,
+            "std": 0.163859
+          },
+          "eb_wse_weak_20_40_snr": {
+            "mean": -10.249423,
+            "std": 0.443729
+          },
+          "fb_fre_high_energy_ratio": {
+            "mean": 0.140025,
+            "std": 0.0
+          },
+          "fb_fre_high_frequency_range_hz": [
+            48.25,
+            73.3
+          ],
+          "fb_fre_high_ne": {
+            "mean": 0.116582,
+            "std": 0.008488
+          },
+          "fb_fre_high_snr": {
+            "mean": 18.733423,
+            "std": 0.654753
+          },
+          "fb_fre_low_energy_ratio": {
+            "mean": 0.125627,
+            "std": 0.0
+          },
+          "fb_fre_low_frequency_range_hz": [
+            6.5,
+            23.2
+          ],
+          "fb_fre_low_ne": {
+            "mean": 0.137702,
+            "std": 0.003074
+          },
+          "fb_fre_low_snr": {
+            "mean": 17.228044,
+            "std": 0.195914
+          },
+          "fb_fre_mid_energy_ratio": {
+            "mean": 0.637894,
+            "std": 0.0
+          },
+          "fb_fre_mid_frequency_range_hz": [
+            23.2,
+            48.25
+          ],
+          "fb_fre_mid_ne": {
+            "mean": 0.073755,
+            "std": 0.002954
+          },
+          "fb_fre_mid_snr": {
+            "mean": 22.651331,
+            "std": 0.35266
+          },
+          "fb_fre_very_high_energy_ratio": {
+            "mean": 0.001545,
+            "std": 0.0
+          },
+          "fb_fre_very_high_frequency_range_hz": [
+            73.3,
+            90.0
+          ],
+          "fb_fre_very_high_ne": {
+            "mean": 0.79646,
+            "std": 0.049241
+          },
+          "fb_fre_very_high_snr": {
+            "mean": 2.090478,
+            "std": 0.549076
+          },
+          "mae": {
+            "mean": 0.004009,
+            "std": 0.000166
+          },
+          "mse": {
+            "mean": 3.8e-05,
+            "std": 4e-06
+          },
+          "psnr": {
+            "mean": 44.3549,
+            "std": 0.415884
+          },
+          "rmse": {
+            "mean": 0.006109,
+            "std": 0.000288
+          },
+          "snr": {
+            "mean": 19.179233,
+            "std": 0.415854
+          },
+          "ssim": {
+            "mean": 0.9863,
+            "std": 0.000648
+          }
+        },
+        "atten_unet": {
+          "eb_wse_medium_40_70_ne": {
+            "mean": 0.851327,
+            "std": 0.007454
+          },
+          "eb_wse_medium_40_70_snr": {
+            "mean": 1.398753,
+            "std": 0.076326
+          },
+          "eb_wse_strong_70_100_ne": {
+            "mean": 0.077606,
+            "std": 0.004972
+          },
+          "eb_wse_strong_70_100_snr": {
+            "mean": 22.22396,
+            "std": 0.541532
+          },
+          "eb_wse_very_weak_5_20_ne": {
+            "mean": 3.095065,
+            "std": 0.207962
+          },
+          "eb_wse_very_weak_5_20_snr": {
+            "mean": -9.785202,
+            "std": 0.587184
+          },
+          "eb_wse_weak_20_40_ne": {
+            "mean": 1.551991,
+            "std": 0.078837
+          },
+          "eb_wse_weak_20_40_snr": {
+            "mean": -3.802846,
+            "std": 0.448202
+          },
+          "fb_fre_high_energy_ratio": {
+            "mean": 0.140025,
+            "std": 0.0
+          },
+          "fb_fre_high_frequency_range_hz": [
+            48.25,
+            73.3
+          ],
+          "fb_fre_high_ne": {
+            "mean": 0.098203,
+            "std": 0.006586
+          },
+          "fb_fre_high_snr": {
+            "mean": 20.195024,
+            "std": 0.575159
+          },
+          "fb_fre_low_energy_ratio": {
+            "mean": 0.125627,
+            "std": 0.0
+          },
+          "fb_fre_low_frequency_range_hz": [
+            6.5,
+            23.2
+          ],
+          "fb_fre_low_ne": {
+            "mean": 0.122955,
+            "std": 0.002681
+          },
+          "fb_fre_low_snr": {
+            "mean": 18.210447,
+            "std": 0.187674
+          },
+          "fb_fre_mid_energy_ratio": {
+            "mean": 0.637894,
+            "std": 0.0
+          },
+          "fb_fre_mid_frequency_range_hz": [
+            23.2,
+            48.25
+          ],
+          "fb_fre_mid_ne": {
+            "mean": 0.070269,
+            "std": 0.003938
+          },
+          "fb_fre_mid_snr": {
+            "mean": 23.07859,
+            "std": 0.47803
+          },
+          "fb_fre_very_high_energy_ratio": {
+            "mean": 0.001545,
+            "std": 0.0
+          },
+          "fb_fre_very_high_frequency_range_hz": [
+            73.3,
+            90.0
+          ],
+          "fb_fre_very_high_ne": {
+            "mean": 0.584715,
+            "std": 0.024681
+          },
+          "fb_fre_very_high_snr": {
+            "mean": 4.73982,
+            "std": 0.371157
+          },
+          "mae": {
+            "mean": 0.003216,
+            "std": 0.000116
+          },
+          "mse": {
+            "mean": 2.8e-05,
+            "std": 3e-06
+          },
+          "psnr": {
+            "mean": 45.7603,
+            "std": 0.408998
+          },
+          "rmse": {
+            "mean": 0.005203,
+            "std": 0.000255
+          },
+          "snr": {
+            "mean": 20.5846,
+            "std": 0.408998
+          },
+          "ssim": {
+            "mean": 0.990033,
+            "std": 0.00045
+          }
+        },
+        "q_unet": {
+          "eb_wse_medium_40_70_ne": {
+            "mean": 0.806676,
+            "std": 0.0086
+          },
+          "eb_wse_medium_40_70_snr": {
+            "mean": 1.867819,
+            "std": 0.091363
+          },
+          "eb_wse_strong_70_100_ne": {
+            "mean": 0.066082,
+            "std": 0.005477
+          },
+          "eb_wse_strong_70_100_snr": {
+            "mean": 23.630982,
+            "std": 0.701696
+          },
+          "eb_wse_very_weak_5_20_ne": {
+            "mean": 2.228961,
+            "std": 0.2573
+          },
+          "eb_wse_very_weak_5_20_snr": {
+            "mean": -6.895089,
+            "std": 1.020328
+          },
+          "eb_wse_weak_20_40_ne": {
+            "mean": 1.284304,
+            "std": 0.077039
+          },
+          "eb_wse_weak_20_40_snr": {
+            "mean": -2.156339,
+            "std": 0.527503
+          },
+          "fb_fre_high_energy_ratio": {
+            "mean": 0.140025,
+            "std": 0.0
+          },
+          "fb_fre_high_frequency_range_hz": [
+            48.25,
+            73.3
+          ],
+          "fb_fre_high_ne": {
+            "mean": 0.079658,
+            "std": 0.006209
+          },
+          "fb_fre_high_snr": {
+            "mean": 22.022328,
+            "std": 0.663156
+          },
+          "fb_fre_low_energy_ratio": {
+            "mean": 0.125627,
+            "std": 0.0
+          },
+          "fb_fre_low_frequency_range_hz": [
+            6.5,
+            23.2
+          ],
+          "fb_fre_low_ne": {
+            "mean": 0.11397,
+            "std": 0.003321
+          },
+          "fb_fre_low_snr": {
+            "mean": 18.870509,
+            "std": 0.253122
+          },
+          "fb_fre_mid_energy_ratio": {
+            "mean": 0.637894,
+            "std": 0.0
+          },
+          "fb_fre_mid_frequency_range_hz": [
+            23.2,
+            48.25
+          ],
+          "fb_fre_mid_ne": {
+            "mean": 0.061408,
+            "std": 0.0017
+          },
+          "fb_fre_mid_snr": {
+            "mean": 24.239222,
+            "std": 0.238276
+          },
+          "fb_fre_very_high_energy_ratio": {
+            "mean": 0.001545,
+            "std": 0.0
+          },
+          "fb_fre_very_high_frequency_range_hz": [
+            73.3,
+            90.0
+          ],
+          "fb_fre_very_high_ne": {
+            "mean": 0.500757,
+            "std": 0.053148
+          },
+          "fb_fre_very_high_snr": {
+            "mean": 6.108169,
+            "std": 0.897716
+          },
+          "mae": {
+            "mean": 0.00276,
+            "std": 7.5e-05
+          },
+          "mse": {
+            "mean": 2.1e-05,
+            "std": 2e-06
+          },
+          "psnr": {
+            "mean": 46.959133,
+            "std": 0.440952
+          },
+          "rmse": {
+            "mean": 0.004533,
+            "std": 0.000236
+          },
+          "snr": {
+            "mean": 21.7835,
+            "std": 0.440998
+          },
+          "ssim": {
+            "mean": 0.9916,
+            "std": 0.000216
+          }
+        },
+        "unet_plusplus": {
+          "eb_wse_medium_40_70_ne": {
+            "mean": 0.757421,
+            "std": 0.028589
+          },
+          "eb_wse_medium_40_70_snr": {
+            "mean": 2.42145,
+            "std": 0.328728
+          },
+          "eb_wse_strong_70_100_ne": {
+            "mean": 0.042676,
+            "std": 0.00108
+          },
+          "eb_wse_strong_70_100_snr": {
+            "mean": 27.403393,
+            "std": 0.223016
+          },
+          "eb_wse_very_weak_5_20_ne": {
+            "mean": 2.291199,
+            "std": 0.019998
+          },
+          "eb_wse_very_weak_5_20_snr": {
+            "mean": -7.179156,
+            "std": 0.074271
+          },
+          "eb_wse_weak_20_40_ne": {
+            "mean": 1.344684,
+            "std": 0.007086
+          },
+          "eb_wse_weak_20_40_snr": {
+            "mean": -2.570046,
+            "std": 0.045083
+          },
+          "fb_fre_high_energy_ratio": {
+            "mean": 0.140025,
+            "std": 0.0
+          },
+          "fb_fre_high_frequency_range_hz": [
+            48.25,
+            73.3
+          ],
+          "fb_fre_high_ne": {
+            "mean": 0.046849,
+            "std": 0.001166
+          },
+          "fb_fre_high_snr": {
+            "mean": 26.628389,
+            "std": 0.222745
+          },
+          "fb_fre_low_energy_ratio": {
+            "mean": 0.125627,
+            "std": 0.0
+          },
+          "fb_fre_low_frequency_range_hz": [
+            6.5,
+            23.2
+          ],
+          "fb_fre_low_ne": {
+            "mean": 0.104241,
+            "std": 0.002996
+          },
+          "fb_fre_low_snr": {
+            "mean": 19.644413,
+            "std": 0.251666
+          },
+          "fb_fre_mid_energy_ratio": {
+            "mean": 0.637894,
+            "std": 0.0
+          },
+          "fb_fre_mid_frequency_range_hz": [
+            23.2,
+            48.25
+          ],
+          "fb_fre_mid_ne": {
+            "mean": 0.05159,
+            "std": 0.001425
+          },
+          "fb_fre_mid_snr": {
+            "mean": 25.752272,
+            "std": 0.240669
+          },
+          "fb_fre_very_high_energy_ratio": {
+            "mean": 0.001545,
+            "std": 0.0
+          },
+          "fb_fre_very_high_frequency_range_hz": [
+            73.3,
+            90.0
+          ],
+          "fb_fre_very_high_ne": {
+            "mean": 0.213581,
+            "std": 0.005819
+          },
+          "fb_fre_very_high_snr": {
+            "mean": 13.517495,
+            "std": 0.24811
+          },
+          "mae": {
+            "mean": 0.002253,
+            "std": 5.7e-05
+          },
+          "mse": {
+            "mean": 1.2e-05,
+            "std": 1e-06
+          },
+          "psnr": {
+            "mean": 49.2956,
+            "std": 0.242266
+          },
+          "rmse": {
+            "mean": 0.003447,
+            "std": 9.7e-05
+          },
+          "snr": {
+            "mean": 24.119933,
+            "std": 0.242313
+          },
+          "ssim": {
+            "mean": 0.9923,
+            "std": 0.000432
+          }
+        },
+        "ddpm": {
+          "eb_wse_medium_40_70_ne": {
+            "mean": 1.385622,
+            "std": 0.080286
+          },
+          "eb_wse_medium_40_70_snr": {
+            "mean": -2.812194,
+            "std": 0.496255
+          },
+          "eb_wse_strong_70_100_ne": {
+            "mean": 0.073466,
+            "std": 0.003703
+          },
+          "eb_wse_strong_70_100_snr": {
+            "mean": 22.717262,
+            "std": 0.42653
+          },
+          "eb_wse_very_weak_5_20_ne": {
+            "mean": 11.00483,
+            "std": 1.23263
+          },
+          "eb_wse_very_weak_5_20_snr": {
+            "mean": -20.735116,
+            "std": 0.930342
+          },
+          "eb_wse_weak_20_40_ne": {
+            "mean": 4.621439,
+            "std": 0.44751
+          },
+          "eb_wse_weak_20_40_snr": {
+            "mean": -13.230891,
+            "std": 0.813904
+          },
+          "fb_fre_high_energy_ratio": {
+            "mean": 0.140025,
+            "std": 0.0
+          },
+          "fb_fre_high_frequency_range_hz": [
+            48.25,
+            73.3
+          ],
+          "fb_fre_high_ne": {
+            "mean": 0.141598,
+            "std": 0.011702
+          },
+          "fb_fre_high_snr": {
+            "mean": 17.185792,
+            "std": 0.681757
+          },
+          "fb_fre_low_energy_ratio": {
+            "mean": 0.125627,
+            "std": 0.0
+          },
+          "fb_fre_low_frequency_range_hz": [
+            6.5,
+            23.2
+          ],
+          "fb_fre_low_ne": {
+            "mean": 0.155503,
+            "std": 0.008703
+          },
+          "fb_fre_low_snr": {
+            "mean": 16.182126,
+            "std": 0.478423
+          },
+          "fb_fre_mid_energy_ratio": {
+            "mean": 0.637894,
+            "std": 0.0
+          },
+          "fb_fre_mid_frequency_range_hz": [
+            23.2,
+            48.25
+          ],
+          "fb_fre_mid_ne": {
+            "mean": 0.081625,
+            "std": 0.004615
+          },
+          "fb_fre_mid_snr": {
+            "mean": 21.790797,
+            "std": 0.479828
+          },
+          "fb_fre_very_high_energy_ratio": {
+            "mean": 0.001545,
+            "std": 0.0
+          },
+          "fb_fre_very_high_frequency_range_hz": [
+            73.3,
+            90.0
+          ],
+          "fb_fre_very_high_ne": {
+            "mean": 1.070203,
+            "std": 0.095554
+          },
+          "fb_fre_very_high_snr": {
+            "mean": -0.277951,
+            "std": 0.734797
+          },
+          "mae": {
+            "mean": 0.005497,
+            "std": 0.000301
+          },
+          "mse": {
+            "mean": 5.4e-05,
+            "std": 8e-06
+          },
+          "psnr": {
+            "mean": 42.7883,
+            "std": 0.613675
+          },
+          "rmse": {
+            "mean": 0.007285,
+            "std": 0.000513
+          },
+          "snr": {
+            "mean": 17.612633,
+            "std": 0.613714
+          },
+          "ssim": {
+            "mean": 0.979367,
+            "std": 0.002237
+          }
+        },
+        "unet_L": {
+          "eb_wse_medium_40_70_ne": {
+            "mean": 0.750301,
+            "std": 0.021546
+          },
+          "eb_wse_medium_40_70_snr": {
+            "mean": 2.500141,
+            "std": 0.246917
+          },
+          "eb_wse_strong_70_100_ne": {
+            "mean": 0.045716,
+            "std": 0.00073
+          },
+          "eb_wse_strong_70_100_snr": {
+            "mean": 26.800401,
+            "std": 0.13896
+          },
+          "eb_wse_very_weak_5_20_ne": {
+            "mean": 2.004984,
+            "std": 0.068431
+          },
+          "eb_wse_very_weak_5_20_snr": {
+            "mean": -6.013512,
+            "std": 0.297801
+          },
+          "eb_wse_weak_20_40_ne": {
+            "mean": 1.226751,
+            "std": 0.019845
+          },
+          "eb_wse_weak_20_40_snr": {
+            "mean": -1.771558,
+            "std": 0.142776
+          },
+          "fb_fre_high_energy_ratio": {
+            "mean": 0.140025,
+            "std": 0.0
+          },
+          "fb_fre_high_frequency_range_hz": [
+            48.25,
+            73.3
+          ],
+          "fb_fre_high_ne": {
+            "mean": 0.05217,
+            "std": 0.001292
+          },
+          "fb_fre_high_snr": {
+            "mean": 25.696429,
+            "std": 0.21256
+          },
+          "fb_fre_low_energy_ratio": {
+            "mean": 0.125627,
+            "std": 0.0
+          },
+          "fb_fre_low_frequency_range_hz": [
+            6.5,
+            23.2
+          ],
+          "fb_fre_low_ne": {
+            "mean": 0.100387,
+            "std": 0.002986
+          },
+          "fb_fre_low_snr": {
+            "mean": 19.971045,
+            "std": 0.256503
+          },
+          "fb_fre_mid_energy_ratio": {
+            "mean": 0.637894,
+            "std": 0.0
+          },
+          "fb_fre_mid_frequency_range_hz": [
+            23.2,
+            48.25
+          ],
+          "fb_fre_mid_ne": {
+            "mean": 0.051987,
+            "std": 0.000848
+          },
+          "fb_fre_mid_snr": {
+            "mean": 25.68345,
+            "std": 0.141003
+          },
+          "fb_fre_very_high_energy_ratio": {
+            "mean": 0.001545,
+            "std": 0.0
+          },
+          "fb_fre_very_high_frequency_range_hz": [
+            73.3,
+            90.0
+          ],
+          "fb_fre_very_high_ne": {
+            "mean": 0.267608,
+            "std": 0.013954
+          },
+          "fb_fre_very_high_snr": {
+            "mean": 11.557521,
+            "std": 0.450459
+          },
+          "mae": {
+            "mean": 0.002287,
+            "std": 2.9e-05
+          },
+          "mse": {
+            "mean": 1.3e-05,
+            "std": 0.0
+          },
+          "psnr": {
+            "mean": 49.105433,
+            "std": 0.13696
+          },
+          "rmse": {
+            "mean": 0.003525,
+            "std": 5.4e-05
+          },
+          "snr": {
+            "mean": 23.929767,
+            "std": 0.137004
+          },
+          "ssim": {
+            "mean": 0.992933,
+            "std": 0.00045
+          }
+        },
+        "SCRN": {
+          "eb_wse_medium_40_70_ne": {
+            "mean": 0.604353,
+            "std": 0.002911
+          },
+          "eb_wse_medium_40_70_snr": {
+            "mean": 4.374492,
+            "std": 0.042071
+          },
+          "eb_wse_strong_70_100_ne": {
+            "mean": 0.038012,
+            "std": 2e-06
+          },
+          "eb_wse_strong_70_100_snr": {
+            "mean": 28.406403,
+            "std": 0.000749
+          },
+          "eb_wse_very_weak_5_20_ne": {
+            "mean": 2.115244,
+            "std": 0.084896
+          },
+          "eb_wse_very_weak_5_20_snr": {
+            "mean": -6.481049,
+            "std": 0.350005
+          },
+          "eb_wse_weak_20_40_ne": {
+            "mean": 1.235941,
+            "std": 0.018991
+          },
+          "eb_wse_weak_20_40_snr": {
+            "mean": -1.835449,
+            "std": 0.133168
+          },
+          "fb_fre_high_energy_ratio": {
+            "mean": 0.140025,
+            "std": 0.0
+          },
+          "fb_fre_high_frequency_range_hz": [
+            48.25,
+            73.3
+          ],
+          "fb_fre_high_ne": {
+            "mean": 0.044643,
+            "std": 0.000415
+          },
+          "fb_fre_high_snr": {
+            "mean": 27.052621,
+            "std": 0.086306
+          },
+          "fb_fre_low_energy_ratio": {
+            "mean": 0.125627,
+            "std": 0.0
+          },
+          "fb_fre_low_frequency_range_hz": [
+            6.5,
+            23.2
+          ],
+          "fb_fre_low_ne": {
+            "mean": 0.08192,
+            "std": 0.000374
+          },
+          "fb_fre_low_snr": {
+            "mean": 21.735244,
+            "std": 0.039803
+          },
+          "fb_fre_mid_energy_ratio": {
+            "mean": 0.637894,
+            "std": 0.0
+          },
+          "fb_fre_mid_frequency_range_hz": [
+            23.2,
+            48.25
+          ],
+          "fb_fre_mid_ne": {
+            "mean": 0.046667,
+            "std": 0.000178
+          },
+          "fb_fre_mid_snr": {
+            "mean": 26.621038,
+            "std": 0.033382
+          },
+          "fb_fre_very_high_energy_ratio": {
+            "mean": 0.001545,
+            "std": 0.0
+          },
+          "fb_fre_very_high_frequency_range_hz": [
+            73.3,
+            90.0
+          ],
+          "fb_fre_very_high_ne": {
+            "mean": 0.166604,
+            "std": 0.000429
+          },
+          "fb_fre_very_high_snr": {
+            "mean": 15.6849,
+            "std": 0.027718
+          },
+          "mae": {
+            "mean": 0.001947,
+            "std": 1e-05
+          },
+          "mse": {
+            "mean": 9e-06,
+            "std": 0.0
+          },
+          "psnr": {
+            "mean": 50.660833,
+            "std": 0.037411
+          },
+          "rmse": {
+            "mean": 0.002941,
+            "std": 1.3e-05
+          },
+          "snr": {
+            "mean": 25.485167,
+            "std": 0.037457
+          },
+          "ssim": {
+            "mean": 0.995333,
+            "std": 4.7e-05
+          }
+        },
+        "ffcnn": {
+          "eb_wse_medium_40_70_ne": {
+            "mean": 1.026906,
+            "std": 0.0
+          },
+          "eb_wse_medium_40_70_snr": {
+            "mean": -0.223421,
+            "std": 0.0
+          },
+          "eb_wse_strong_70_100_ne": {
+            "mean": 0.308648,
+            "std": 0.0
+          },
+          "eb_wse_strong_70_100_snr": {
+            "mean": 10.213956,
+            "std": 0.0
+          },
+          "eb_wse_very_weak_5_20_ne": {
+            "mean": 3.212355,
+            "std": 0.0
+          },
+          "eb_wse_very_weak_5_20_snr": {
+            "mean": -10.11961,
+            "std": 0.0
+          },
+          "eb_wse_weak_20_40_ne": {
+            "mean": 1.800039,
+            "std": 0.0
+          },
+          "eb_wse_weak_20_40_snr": {
+            "mean": -5.090044,
+            "std": 0.0
+          },
+          "fb_fre_high_energy_ratio": {
+            "mean": 0.140025,
+            "std": 0.0
+          },
+          "fb_fre_high_frequency_range_hz": [
+            48.25,
+            73.3
+          ],
+          "fb_fre_high_ne": {
+            "mean": 0.477545,
+            "std": 0.0
+          },
+          "fb_fre_high_snr": {
+            "mean": 6.429011,
+            "std": 0.0
+          },
+          "fb_fre_low_energy_ratio": {
+            "mean": 0.125627,
+            "std": 0.0
+          },
+          "fb_fre_low_frequency_range_hz": [
+            6.5,
+            23.2
+          ],
+          "fb_fre_low_ne": {
+            "mean": 0.262586,
+            "std": 0.0
+          },
+          "fb_fre_low_snr": {
+            "mean": 11.621104,
+            "std": 0.0
+          },
+          "fb_fre_mid_energy_ratio": {
+            "mean": 0.637894,
+            "std": 0.0
+          },
+          "fb_fre_mid_frequency_range_hz": [
+            23.2,
+            48.25
+          ],
+          "fb_fre_mid_ne": {
+            "mean": 0.245133,
+            "std": 0.0
+          },
+          "fb_fre_mid_snr": {
+            "mean": 12.212883,
+            "std": 0.0
+          },
+          "fb_fre_very_high_energy_ratio": {
+            "mean": 0.001545,
+            "std": 0.0
+          },
+          "fb_fre_very_high_frequency_range_hz": [
+            73.3,
+            90.0
+          ],
+          "fb_fre_very_high_ne": {
+            "mean": 1.570027,
+            "std": 0.0
+          },
+          "fb_fre_very_high_snr": {
+            "mean": -3.882686,
+            "std": 0.0
+          },
+          "mae": {
+            "mean": 0.007205,
+            "std": 0.0
+          },
+          "mse": {
+            "mean": 0.000314,
+            "std": 0.0
+          },
+          "psnr": {
+            "mean": 35.2311,
+            "std": 0.0
+          },
+          "rmse": {
+            "mean": 0.017515,
+            "std": 0.0
+          },
+          "snr": {
+            "mean": 10.0554,
+            "std": 0.0
+          },
+          "ssim": {
+            "mean": 0.967,
+            "std": 0.0
+          }
+        },
+        "fbresnet": {
+          "eb_wse_medium_40_70_ne": {
+            "mean": 1.040688,
+            "std": 0.0
+          },
+          "eb_wse_medium_40_70_snr": {
+            "mean": -0.344219,
+            "std": 0.0
+          },
+          "eb_wse_strong_70_100_ne": {
+            "mean": 0.052394,
+            "std": 0.0
+          },
+          "eb_wse_strong_70_100_snr": {
+            "mean": 25.614623,
+            "std": 0.0
+          },
+          "eb_wse_very_weak_5_20_ne": {
+            "mean": 6.730613,
+            "std": 0.0
+          },
+          "eb_wse_very_weak_5_20_snr": {
+            "mean": -16.543804,
+            "std": 0.0
+          },
+          "eb_wse_weak_20_40_ne": {
+            "mean": 3.019342,
+            "std": 0.0
+          },
+          "eb_wse_weak_20_40_snr": {
+            "mean": -9.583882,
+            "std": 0.0
+          },
+          "fb_fre_high_energy_ratio": {
+            "mean": 0.140025,
+            "std": 0.0
+          },
+          "fb_fre_high_frequency_range_hz": [
+            48.25,
+            73.3
+          ],
+          "fb_fre_high_ne": {
+            "mean": 0.090167,
+            "std": 0.0
+          },
+          "fb_fre_high_snr": {
+            "mean": 20.937571,
+            "std": 0.0
+          },
+          "fb_fre_low_energy_ratio": {
+            "mean": 0.125627,
+            "std": 0.0
+          },
+          "fb_fre_low_frequency_range_hz": [
+            6.5,
+            23.2
+          ],
+          "fb_fre_low_ne": {
+            "mean": 0.122403,
+            "std": 0.0
+          },
+          "fb_fre_low_snr": {
+            "mean": 18.245308,
+            "std": 0.0
+          },
+          "fb_fre_mid_energy_ratio": {
+            "mean": 0.637894,
+            "std": 0.0
+          },
+          "fb_fre_mid_frequency_range_hz": [
+            23.2,
+            48.25
+          ],
+          "fb_fre_mid_ne": {
+            "mean": 0.070533,
+            "std": 0.0
+          },
+          "fb_fre_mid_snr": {
+            "mean": 23.032445,
+            "std": 0.0
+          },
+          "fb_fre_very_high_energy_ratio": {
+            "mean": 0.001545,
+            "std": 0.0
+          },
+          "fb_fre_very_high_frequency_range_hz": [
+            73.3,
+            90.0
+          ],
+          "fb_fre_very_high_ne": {
+            "mean": 0.502504,
+            "std": 0.0
+          },
+          "fb_fre_very_high_snr": {
+            "mean": 6.0915,
+            "std": 0.0
+          },
+          "mae": {
+            "mean": 0.003269,
+            "std": 0.0
+          },
+          "mse": {
+            "mean": 2.5e-05,
+            "std": 0.0
+          },
+          "psnr": {
+            "mean": 46.0525,
+            "std": 0.0
+          },
+          "rmse": {
+            "mean": 0.005014,
+            "std": 0.0
+          },
+          "snr": {
+            "mean": 20.8769,
+            "std": 0.0
+          },
+          "ssim": {
+            "mean": 0.9882,
+            "std": 0.0
+          }
+        }
+      }
+    }
+  },
+  "num_params": {
+    "unet": 7760000,
+    "dncnn": 560000,
+    "res_unet": 8110000,
+    "atten_unet": 7850000,
+    "q_unet": 21880000,
+    "unet_plusplus": 9050000,
+    "ddpm": 33300000,
+    "unet_L": 31040000,
+    "SCRN": 430000,
+    "ffcnn": 560000,
+    "fbresnet": 450000
+  }
+}
+AUTO_RESULTS_DATA_END -->
+<!-- AUTO_RESULTS_END -->

blending-noise-suppression-avo/models/SCRN/T03_avo_mod_seed42/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:22e7e533f20153ac3809c1e8134f521b175bf462137c499ac78451ed3061acd8
+size 4874136

blending-noise-suppression-avo/models/SCRN/T03_avo_mod_seed42/config.yaml

@@ -0,0 +1,132 @@
+experiment:
+  name: blending_noise_avo_SCRN_base_T03_avo_mod_seed42_T03_pseudo_deblended_seismic_common_receiver_mod
+  output_dir: /data/data/results/blending_noise_avo/SCRN
+  seed: 42
+  device: cuda
+data:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  loader:
+    batch_size: 76
+    num_workers: 4
+    pin_memory: true
+preprocess:
+  dt: 0.004
+  normalize_mode: max_abs
+  normalize_scope: shot
+  patch_time: 256
+  patch_trace: 128
+  patch_overlap: 0.5
+  max_gathers: null
+  skip: []
+model:
+  type: SCRN
+  params:
+    in_nc: 1
+    config:
+    - 1
+    - 1
+    - 1
+    - 1
+    - 1
+    dim: 64
+    drop_path_rate: 0.0
+    input_resolution: 128
+loss:
+  type: mse
+  params:
+    reduction: mean
+metrics:
+- name: snr
+  params:
+    reduction: per_sample
+- name: psnr
+  params:
+    data_range: 1.0
+    reduction: per_sample
+- name: ssim
+  params:
+    data_range: 2.0
+    window_size: 11
+    sigma: 1.5
+- name: mae
+  params: {}
+- name: mse
+  params: {}
+- name: rmse
+  params:
+    reduction: per_sample
+metrics_filter:
+  snr_min_signal_energy: 1.0e-08
+optim:
+  type: adamw
+  params:
+    lr: 0.0001
+    weight_decay: 1.0e-05
+scheduler:
+  type: cosine
+  params:
+    min_lr: 1.0e-06
+train:
+  epochs: 200
+  grad_clip: 1.0
+  log_step: false
+  log_interval: 10
+  eval_interval: 1
+  ckpt_interval: 20
+  vis_interval: 5
+  resume: null
+log:
+  log_dir: logs
+  plot_interval: 5
+inference:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  checkpoint: /data/data/results/blending_noise_avo/SCRN/blending_noise_avo_SCRN_base/checkpoints/best.pt
+  output_dir: /data/data/results/blending_noise_avo/SCRN/blending_noise_avo_SCRN_base/inference_integrated_metrics
+  n_viz_gathers: 5
+  device: cuda:0
+  batch_size: 76
+  save_npy: false
+  binned_metrics:
+    enabled: true
+    eb_wse:
+      enabled: true
+      bins:
+      - - 5
+        - 20
+      - - 20
+        - 40
+      - - 40
+        - 70
+      - - 70
+        - 100
+      smooth_sigma: 1.0
+    fb_fre:
+      enabled: true
+      rel_threshold: 0.001
+      band_ratios:
+      - 0.2
+      - 0.3
+      - 0.3
+      - 0.2
+      band_names:
+      - low
+      - mid
+      - high
+      - very_high
+      taper_width: null

blending-noise-suppression-avo/models/SCRN/T03_avo_mod_seed42/metrics_summary.json

@@ -0,0 +1,170 @@
+{
+  "task": "blending_noise_suppression",
+  "sample_kind": "gather",
+  "sample_axis_name": "common_receiver_gather",
+  "input": {
+    "snr": -2.0292,
+    "psnr": 23.1464,
+    "ssim": 0.7028,
+    "mae": 0.025431,
+    "mse": 0.005028636660426855,
+    "rmse": 0.070259,
+    "eb_wse_very_weak_5_20_ne": 131.715259,
+    "eb_wse_very_weak_5_20_snr": -42.35256,
+    "eb_wse_weak_20_40_ne": 54.756158,
+    "eb_wse_weak_20_40_snr": -34.743819,
+    "eb_wse_medium_40_70_ne": 16.207542,
+    "eb_wse_medium_40_70_snr": -24.181704,
+    "eb_wse_strong_70_100_ne": 0.211915,
+    "eb_wse_strong_70_100_snr": 13.511556,
+    "fb_fre_low_ne": 1.251054,
+    "fb_fre_low_snr": -1.945173,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 1.265097,
+    "fb_fre_mid_snr": -2.042182,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 1.263001,
+    "fb_fre_high_snr": -2.027766,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 1.300019,
+    "fb_fre_very_high_snr": -2.278716,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "denoised": {
+    "snr": 25.4696,
+    "psnr": 50.6453,
+    "ssim": 0.9953,
+    "mae": 0.00195,
+    "mse": 8.750824235903565e-06,
+    "rmse": 0.002947,
+    "eb_wse_very_weak_5_20_ne": 2.135524,
+    "eb_wse_very_weak_5_20_snr": -6.561983,
+    "eb_wse_weak_20_40_ne": 1.240717,
+    "eb_wse_weak_20_40_snr": -1.868246,
+    "eb_wse_medium_40_70_ne": 0.606155,
+    "eb_wse_medium_40_70_snr": 4.348424,
+    "eb_wse_strong_70_100_ne": 0.038014,
+    "eb_wse_strong_70_100_snr": 28.405546,
+    "fb_fre_low_ne": 0.08191,
+    "fb_fre_low_snr": 21.737013,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.046682,
+    "fb_fre_mid_snr": 26.617719,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.045177,
+    "fb_fre_high_snr": 26.942319,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 0.167156,
+    "fb_fre_very_high_snr": 15.650203,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "delta": {
+    "snr": 27.49884,
+    "psnr": 27.49884,
+    "ssim": 0.292545,
+    "mae": -0.023481,
+    "mse": -0.00502,
+    "rmse": -0.067312,
+    "eb_wse_very_weak_5_20_ne": -129.579735,
+    "eb_wse_very_weak_5_20_snr": 35.790577,
+    "eb_wse_weak_20_40_ne": -53.515441,
+    "eb_wse_weak_20_40_snr": 32.875573,
+    "eb_wse_medium_40_70_ne": -15.601387,
+    "eb_wse_medium_40_70_snr": 28.530128,
+    "eb_wse_strong_70_100_ne": -0.173901,
+    "eb_wse_strong_70_100_snr": 14.89399,
+    "fb_fre_low_ne": -1.169144,
+    "fb_fre_low_snr": 23.682186,
+    "fb_fre_mid_ne": -1.218415,
+    "fb_fre_mid_snr": 28.659901,
+    "fb_fre_high_ne": -1.217824,
+    "fb_fre_high_snr": 28.970085,
+    "fb_fre_very_high_ne": -1.132863,
+    "fb_fre_very_high_snr": 17.928919
+  },
+  "input_path": "/data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy",
+  "target_path": "/data/data/T03_seismic_common_receiver.sgy",
+  "input_source": "paired_pseudo_deblended_common_receiver",
+  "inference_time_seconds": 8.848,
+  "num_test_gathers": 12,
+  "snr_min_signal_energy": 1e-08,
+  "input_snr_valid_gathers": 12,
+  "denoised_snr_valid_gathers": 12,
+  "snr_total_gathers": 12,
+  "restored": {
+    "snr": 25.4696,
+    "psnr": 50.6453,
+    "ssim": 0.9953,
+    "mae": 0.00195,
+    "mse": 8.750824235903565e-06,
+    "rmse": 0.002947,
+    "eb_wse_very_weak_5_20_ne": 2.135524,
+    "eb_wse_very_weak_5_20_snr": -6.561983,
+    "eb_wse_weak_20_40_ne": 1.240717,
+    "eb_wse_weak_20_40_snr": -1.868246,
+    "eb_wse_medium_40_70_ne": 0.606155,
+    "eb_wse_medium_40_70_snr": 4.348424,
+    "eb_wse_strong_70_100_ne": 0.038014,
+    "eb_wse_strong_70_100_snr": 28.405546,
+    "fb_fre_low_ne": 0.08191,
+    "fb_fre_low_snr": 21.737013,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.046682,
+    "fb_fre_mid_snr": 26.617719,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.045177,
+    "fb_fre_high_snr": 26.942319,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 0.167156,
+    "fb_fre_very_high_snr": 15.650203,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  }
+}

blending-noise-suppression-avo/models/SCRN/T03_avo_mod_seed42/model_metadata.json

@@ -0,0 +1,7 @@
+{
+  "task": "blending_noise_suppression",
+  "model": "SCRN",
+  "variant": "T03_avo_mod",
+  "seed": "42",
+  "num_parameters": 430000
+}

blending-noise-suppression-avo/models/SCRN/T03_avo_mod_seed43/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:51eb7e14adc4a55d514ac6c746ca35469f69b687dd8509b78bd0510b657661ac
+size 4873303

blending-noise-suppression-avo/models/SCRN/T03_avo_mod_seed43/config.yaml

@@ -0,0 +1,132 @@
+experiment:
+  name: blending_noise_avo_SCRN_base_T03_avo_mod_seed43_T03_pseudo_deblended_seismic_common_receiver_mod
+  output_dir: /data/data/results/blending_noise_avo/SCRN
+  seed: 43
+  device: cuda
+data:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  loader:
+    batch_size: 76
+    num_workers: 4
+    pin_memory: true
+preprocess:
+  dt: 0.004
+  normalize_mode: max_abs
+  normalize_scope: shot
+  patch_time: 256
+  patch_trace: 128
+  patch_overlap: 0.5
+  max_gathers: null
+  skip: []
+model:
+  type: SCRN
+  params:
+    in_nc: 1
+    config:
+    - 1
+    - 1
+    - 1
+    - 1
+    - 1
+    dim: 64
+    drop_path_rate: 0.0
+    input_resolution: 128
+loss:
+  type: mse
+  params:
+    reduction: mean
+metrics:
+- name: snr
+  params:
+    reduction: per_sample
+- name: psnr
+  params:
+    data_range: 1.0
+    reduction: per_sample
+- name: ssim
+  params:
+    data_range: 2.0
+    window_size: 11
+    sigma: 1.5
+- name: mae
+  params: {}
+- name: mse
+  params: {}
+- name: rmse
+  params:
+    reduction: per_sample
+metrics_filter:
+  snr_min_signal_energy: 1.0e-08
+optim:
+  type: adamw
+  params:
+    lr: 0.0001
+    weight_decay: 1.0e-05
+scheduler:
+  type: cosine
+  params:
+    min_lr: 1.0e-06
+train:
+  epochs: 200
+  grad_clip: 1.0
+  log_step: false
+  log_interval: 10
+  eval_interval: 1
+  ckpt_interval: 20
+  vis_interval: 5
+  resume: null
+log:
+  log_dir: logs
+  plot_interval: 5
+inference:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  checkpoint: /data/data/results/blending_noise_avo/SCRN/blending_noise_avo_SCRN_base/checkpoints/best.pt
+  output_dir: /data/data/results/blending_noise_avo/SCRN/blending_noise_avo_SCRN_base/inference_integrated_metrics
+  n_viz_gathers: 5
+  device: cuda:0
+  batch_size: 76
+  save_npy: false
+  binned_metrics:
+    enabled: true
+    eb_wse:
+      enabled: true
+      bins:
+      - - 5
+        - 20
+      - - 20
+        - 40
+      - - 40
+        - 70
+      - - 70
+        - 100
+      smooth_sigma: 1.0
+    fb_fre:
+      enabled: true
+      rel_threshold: 0.001
+      band_ratios:
+      - 0.2
+      - 0.3
+      - 0.3
+      - 0.2
+      band_names:
+      - low
+      - mid
+      - high
+      - very_high
+      taper_width: null

blending-noise-suppression-avo/models/SCRN/T03_avo_mod_seed43/metrics_summary.json

@@ -0,0 +1,170 @@
+{
+  "task": "blending_noise_suppression",
+  "sample_kind": "gather",
+  "sample_axis_name": "common_receiver_gather",
+  "input": {
+    "snr": -2.0292,
+    "psnr": 23.1464,
+    "ssim": 0.7028,
+    "mae": 0.025431,
+    "mse": 0.005028636660426855,
+    "rmse": 0.070259,
+    "eb_wse_very_weak_5_20_ne": 131.715259,
+    "eb_wse_very_weak_5_20_snr": -42.35256,
+    "eb_wse_weak_20_40_ne": 54.756158,
+    "eb_wse_weak_20_40_snr": -34.743819,
+    "eb_wse_medium_40_70_ne": 16.207542,
+    "eb_wse_medium_40_70_snr": -24.181704,
+    "eb_wse_strong_70_100_ne": 0.211915,
+    "eb_wse_strong_70_100_snr": 13.511556,
+    "fb_fre_low_ne": 1.251054,
+    "fb_fre_low_snr": -1.945173,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 1.265097,
+    "fb_fre_mid_snr": -2.042182,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 1.263001,
+    "fb_fre_high_snr": -2.027766,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 1.300019,
+    "fb_fre_very_high_snr": -2.278716,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "denoised": {
+    "snr": 25.4491,
+    "psnr": 50.6248,
+    "ssim": 0.9953,
+    "mae": 0.001958,
+    "mse": 8.785338650341146e-06,
+    "rmse": 0.002953,
+    "eb_wse_very_weak_5_20_ne": 2.207586,
+    "eb_wse_very_weak_5_20_snr": -6.86348,
+    "eb_wse_weak_20_40_ne": 1.256441,
+    "eb_wse_weak_20_40_snr": -1.979655,
+    "eb_wse_medium_40_70_ne": 0.606658,
+    "eb_wse_medium_40_70_snr": 4.341209,
+    "eb_wse_strong_70_100_ne": 0.03801,
+    "eb_wse_strong_70_100_snr": 28.40737,
+    "fb_fre_low_ne": 0.082383,
+    "fb_fre_low_snr": 21.685635,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.046878,
+    "fb_fre_mid_snr": 26.581914,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.044587,
+    "fb_fre_high_snr": 27.062515,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 0.166547,
+    "fb_fre_very_high_snr": 15.686451,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "delta": {
+    "snr": 27.478378,
+    "psnr": 27.478378,
+    "ssim": 0.292481,
+    "mae": -0.023473,
+    "mse": -0.00502,
+    "rmse": -0.067306,
+    "eb_wse_very_weak_5_20_ne": -129.507673,
+    "eb_wse_very_weak_5_20_snr": 35.48908,
+    "eb_wse_weak_20_40_ne": -53.499717,
+    "eb_wse_weak_20_40_snr": 32.764164,
+    "eb_wse_medium_40_70_ne": -15.600884,
+    "eb_wse_medium_40_70_snr": 28.522913,
+    "eb_wse_strong_70_100_ne": -0.173905,
+    "eb_wse_strong_70_100_snr": 14.895814,
+    "fb_fre_low_ne": -1.168671,
+    "fb_fre_low_snr": 23.630808,
+    "fb_fre_mid_ne": -1.218219,
+    "fb_fre_mid_snr": 28.624096,
+    "fb_fre_high_ne": -1.218414,
+    "fb_fre_high_snr": 29.090281,
+    "fb_fre_very_high_ne": -1.133472,
+    "fb_fre_very_high_snr": 17.965167
+  },
+  "input_path": "/data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy",
+  "target_path": "/data/data/T03_seismic_common_receiver.sgy",
+  "input_source": "paired_pseudo_deblended_common_receiver",
+  "inference_time_seconds": 8.701,
+  "num_test_gathers": 12,
+  "snr_min_signal_energy": 1e-08,
+  "input_snr_valid_gathers": 12,
+  "denoised_snr_valid_gathers": 12,
+  "snr_total_gathers": 12,
+  "restored": {
+    "snr": 25.4491,
+    "psnr": 50.6248,
+    "ssim": 0.9953,
+    "mae": 0.001958,
+    "mse": 8.785338650341146e-06,
+    "rmse": 0.002953,
+    "eb_wse_very_weak_5_20_ne": 2.207586,
+    "eb_wse_very_weak_5_20_snr": -6.86348,
+    "eb_wse_weak_20_40_ne": 1.256441,
+    "eb_wse_weak_20_40_snr": -1.979655,
+    "eb_wse_medium_40_70_ne": 0.606658,
+    "eb_wse_medium_40_70_snr": 4.341209,
+    "eb_wse_strong_70_100_ne": 0.03801,
+    "eb_wse_strong_70_100_snr": 28.40737,
+    "fb_fre_low_ne": 0.082383,
+    "fb_fre_low_snr": 21.685635,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.046878,
+    "fb_fre_mid_snr": 26.581914,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.044587,
+    "fb_fre_high_snr": 27.062515,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 0.166547,
+    "fb_fre_very_high_snr": 15.686451,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  }
+}

blending-noise-suppression-avo/models/SCRN/T03_avo_mod_seed43/model_metadata.json

@@ -0,0 +1,7 @@
+{
+  "task": "blending_noise_suppression",
+  "model": "SCRN",
+  "variant": "T03_avo_mod",
+  "seed": "43",
+  "num_parameters": 430000
+}

blending-noise-suppression-avo/models/SCRN/T03_avo_mod_seed44/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0b242a414473f91f6cd146fb02fc5b1fe72e9f6e1b3cbd805602c9cfb2787d81
+size 4874274

blending-noise-suppression-avo/models/SCRN/T03_avo_mod_seed44/config.yaml

@@ -0,0 +1,132 @@
+experiment:
+  name: blending_noise_avo_SCRN_base_T03_avo_mod_seed44_T03_pseudo_deblended_seismic_common_receiver_mod
+  output_dir: /data/data/results/blending_noise_avo/SCRN
+  seed: 44
+  device: cuda
+data:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  loader:
+    batch_size: 76
+    num_workers: 4
+    pin_memory: true
+preprocess:
+  dt: 0.004
+  normalize_mode: max_abs
+  normalize_scope: shot
+  patch_time: 256
+  patch_trace: 128
+  patch_overlap: 0.5
+  max_gathers: null
+  skip: []
+model:
+  type: SCRN
+  params:
+    in_nc: 1
+    config:
+    - 1
+    - 1
+    - 1
+    - 1
+    - 1
+    dim: 64
+    drop_path_rate: 0.0
+    input_resolution: 128
+loss:
+  type: mse
+  params:
+    reduction: mean
+metrics:
+- name: snr
+  params:
+    reduction: per_sample
+- name: psnr
+  params:
+    data_range: 1.0
+    reduction: per_sample
+- name: ssim
+  params:
+    data_range: 2.0
+    window_size: 11
+    sigma: 1.5
+- name: mae
+  params: {}
+- name: mse
+  params: {}
+- name: rmse
+  params:
+    reduction: per_sample
+metrics_filter:
+  snr_min_signal_energy: 1.0e-08
+optim:
+  type: adamw
+  params:
+    lr: 0.0001
+    weight_decay: 1.0e-05
+scheduler:
+  type: cosine
+  params:
+    min_lr: 1.0e-06
+train:
+  epochs: 200
+  grad_clip: 1.0
+  log_step: false
+  log_interval: 10
+  eval_interval: 1
+  ckpt_interval: 20
+  vis_interval: 5
+  resume: null
+log:
+  log_dir: logs
+  plot_interval: 5
+inference:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  checkpoint: /data/data/results/blending_noise_avo/SCRN/blending_noise_avo_SCRN_base/checkpoints/best.pt
+  output_dir: /data/data/results/blending_noise_avo/SCRN/blending_noise_avo_SCRN_base/inference_integrated_metrics
+  n_viz_gathers: 5
+  device: cuda:0
+  batch_size: 76
+  save_npy: false
+  binned_metrics:
+    enabled: true
+    eb_wse:
+      enabled: true
+      bins:
+      - - 5
+        - 20
+      - - 20
+        - 40
+      - - 40
+        - 70
+      - - 70
+        - 100
+      smooth_sigma: 1.0
+    fb_fre:
+      enabled: true
+      rel_threshold: 0.001
+      band_ratios:
+      - 0.2
+      - 0.3
+      - 0.3
+      - 0.2
+      band_names:
+      - low
+      - mid
+      - high
+      - very_high
+      taper_width: null

blending-noise-suppression-avo/models/SCRN/T03_avo_mod_seed44/metrics_summary.json

@@ -0,0 +1,170 @@
+{
+  "task": "blending_noise_suppression",
+  "sample_kind": "gather",
+  "sample_axis_name": "common_receiver_gather",
+  "input": {
+    "snr": -2.0292,
+    "psnr": 23.1464,
+    "ssim": 0.7028,
+    "mae": 0.025431,
+    "mse": 0.005028636660426855,
+    "rmse": 0.070259,
+    "eb_wse_very_weak_5_20_ne": 131.715259,
+    "eb_wse_very_weak_5_20_snr": -42.35256,
+    "eb_wse_weak_20_40_ne": 54.756158,
+    "eb_wse_weak_20_40_snr": -34.743819,
+    "eb_wse_medium_40_70_ne": 16.207542,
+    "eb_wse_medium_40_70_snr": -24.181704,
+    "eb_wse_strong_70_100_ne": 0.211915,
+    "eb_wse_strong_70_100_snr": 13.511556,
+    "fb_fre_low_ne": 1.251054,
+    "fb_fre_low_snr": -1.945173,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 1.265097,
+    "fb_fre_mid_snr": -2.042182,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 1.263001,
+    "fb_fre_high_snr": -2.027766,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 1.300019,
+    "fb_fre_very_high_snr": -2.278716,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "denoised": {
+    "snr": 25.5368,
+    "psnr": 50.7124,
+    "ssim": 0.9954,
+    "mae": 0.001934,
+    "mse": 8.602981324656866e-06,
+    "rmse": 0.002923,
+    "eb_wse_very_weak_5_20_ne": 2.002622,
+    "eb_wse_very_weak_5_20_snr": -6.017685,
+    "eb_wse_weak_20_40_ne": 1.210665,
+    "eb_wse_weak_20_40_snr": -1.658446,
+    "eb_wse_medium_40_70_ne": 0.600246,
+    "eb_wse_medium_40_70_snr": 4.433844,
+    "eb_wse_strong_70_100_ne": 0.038013,
+    "eb_wse_strong_70_100_snr": 28.406294,
+    "fb_fre_low_ne": 0.081467,
+    "fb_fre_low_snr": 21.783084,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.046442,
+    "fb_fre_mid_snr": 26.66348,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.044164,
+    "fb_fre_high_snr": 27.15303,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 0.16611,
+    "fb_fre_very_high_snr": 15.718046,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "delta": {
+    "snr": 27.56601,
+    "psnr": 27.566015,
+    "ssim": 0.292636,
+    "mae": -0.023497,
+    "mse": -0.00502,
+    "rmse": -0.067336,
+    "eb_wse_very_weak_5_20_ne": -129.712637,
+    "eb_wse_very_weak_5_20_snr": 36.334875,
+    "eb_wse_weak_20_40_ne": -53.545493,
+    "eb_wse_weak_20_40_snr": 33.085373,
+    "eb_wse_medium_40_70_ne": -15.607296,
+    "eb_wse_medium_40_70_snr": 28.615548,
+    "eb_wse_strong_70_100_ne": -0.173902,
+    "eb_wse_strong_70_100_snr": 14.894738,
+    "fb_fre_low_ne": -1.169587,
+    "fb_fre_low_snr": 23.728257,
+    "fb_fre_mid_ne": -1.218655,
+    "fb_fre_mid_snr": 28.705662,
+    "fb_fre_high_ne": -1.218837,
+    "fb_fre_high_snr": 29.180796,
+    "fb_fre_very_high_ne": -1.133909,
+    "fb_fre_very_high_snr": 17.996762
+  },
+  "input_path": "/data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy",
+  "target_path": "/data/data/T03_seismic_common_receiver.sgy",
+  "input_source": "paired_pseudo_deblended_common_receiver",
+  "inference_time_seconds": 8.809,
+  "num_test_gathers": 12,
+  "snr_min_signal_energy": 1e-08,
+  "input_snr_valid_gathers": 12,
+  "denoised_snr_valid_gathers": 12,
+  "snr_total_gathers": 12,
+  "restored": {
+    "snr": 25.5368,
+    "psnr": 50.7124,
+    "ssim": 0.9954,
+    "mae": 0.001934,
+    "mse": 8.602981324656866e-06,
+    "rmse": 0.002923,
+    "eb_wse_very_weak_5_20_ne": 2.002622,
+    "eb_wse_very_weak_5_20_snr": -6.017685,
+    "eb_wse_weak_20_40_ne": 1.210665,
+    "eb_wse_weak_20_40_snr": -1.658446,
+    "eb_wse_medium_40_70_ne": 0.600246,
+    "eb_wse_medium_40_70_snr": 4.433844,
+    "eb_wse_strong_70_100_ne": 0.038013,
+    "eb_wse_strong_70_100_snr": 28.406294,
+    "fb_fre_low_ne": 0.081467,
+    "fb_fre_low_snr": 21.783084,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.046442,
+    "fb_fre_mid_snr": 26.66348,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.044164,
+    "fb_fre_high_snr": 27.15303,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 0.16611,
+    "fb_fre_very_high_snr": 15.718046,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  }
+}

blending-noise-suppression-avo/models/SCRN/T03_avo_mod_seed44/model_metadata.json

@@ -0,0 +1,7 @@
+{
+  "task": "blending_noise_suppression",
+  "model": "SCRN",
+  "variant": "T03_avo_mod",
+  "seed": "44",
+  "num_parameters": 430000
+}

blending-noise-suppression-avo/models/atten_unet/T03_avo_mod_seed42/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:01b603b62c216542e5f7bdfe60565d639a56e92169683137fabb121735cc9c84
+size 86911686

blending-noise-suppression-avo/models/atten_unet/T03_avo_mod_seed42/config.yaml

@@ -0,0 +1,126 @@
+experiment:
+  name: blending_noise_avo_atten_unet_base_T03_avo_mod_seed42_T03_pseudo_deblended_seismic_common_receiver_mod
+  output_dir: /data/data/results/blending_noise_avo/atten_unet
+  seed: 42
+  device: cuda
+data:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  loader:
+    batch_size: 360
+    num_workers: 4
+    pin_memory: true
+preprocess:
+  dt: 0.004
+  normalize_mode: max_abs
+  normalize_scope: shot
+  patch_time: 256
+  patch_trace: 128
+  patch_overlap: 0.5
+  max_gathers: null
+  skip: []
+model:
+  type: atten_unet
+  params:
+    in_channels: 1
+    out_channels: 1
+    base_channels: 32
+    depth: 4
+loss:
+  type: mse
+  params:
+    reduction: mean
+metrics:
+- name: snr
+  params:
+    reduction: per_sample
+- name: psnr
+  params:
+    data_range: 1.0
+    reduction: per_sample
+- name: ssim
+  params:
+    data_range: 2.0
+    window_size: 11
+    sigma: 1.5
+- name: mae
+  params: {}
+- name: mse
+  params: {}
+- name: rmse
+  params:
+    reduction: per_sample
+metrics_filter:
+  snr_min_signal_energy: 1.0e-08
+optim:
+  type: adamw
+  params:
+    lr: 0.0001
+    weight_decay: 1.0e-05
+scheduler:
+  type: cosine
+  params:
+    min_lr: 1.0e-06
+train:
+  epochs: 200
+  grad_clip: 1.0
+  log_step: false
+  log_interval: 10
+  eval_interval: 1
+  ckpt_interval: 20
+  vis_interval: 5
+  resume: null
+log:
+  log_dir: logs
+  plot_interval: 5
+inference:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  checkpoint: /data/data/results/blending_noise_avo/atten_unet/blending_noise_avo_atten_unet_base/checkpoints/best.pt
+  output_dir: /data/data/results/blending_noise_avo/atten_unet/blending_noise_avo_atten_unet_base/inference_integrated_metrics
+  n_viz_gathers: 5
+  device: cuda:0
+  batch_size: 360
+  save_npy: false
+  binned_metrics:
+    enabled: true
+    eb_wse:
+      enabled: true
+      bins:
+      - - 5
+        - 20
+      - - 20
+        - 40
+      - - 40
+        - 70
+      - - 70
+        - 100
+      smooth_sigma: 1.0
+    fb_fre:
+      enabled: true
+      rel_threshold: 0.001
+      band_ratios:
+      - 0.2
+      - 0.3
+      - 0.3
+      - 0.2
+      band_names:
+      - low
+      - mid
+      - high
+      - very_high
+      taper_width: null

blending-noise-suppression-avo/models/atten_unet/T03_avo_mod_seed42/metrics_summary.json

@@ -0,0 +1,170 @@
+{
+  "task": "blending_noise_suppression",
+  "sample_kind": "gather",
+  "sample_axis_name": "common_receiver_gather",
+  "input": {
+    "snr": -2.0292,
+    "psnr": 23.1464,
+    "ssim": 0.7028,
+    "mae": 0.025431,
+    "mse": 0.005028636660426855,
+    "rmse": 0.070259,
+    "eb_wse_very_weak_5_20_ne": 131.715259,
+    "eb_wse_very_weak_5_20_snr": -42.35256,
+    "eb_wse_weak_20_40_ne": 54.756158,
+    "eb_wse_weak_20_40_snr": -34.743819,
+    "eb_wse_medium_40_70_ne": 16.207542,
+    "eb_wse_medium_40_70_snr": -24.181704,
+    "eb_wse_strong_70_100_ne": 0.211915,
+    "eb_wse_strong_70_100_snr": 13.511556,
+    "fb_fre_low_ne": 1.251054,
+    "fb_fre_low_snr": -1.945173,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 1.265097,
+    "fb_fre_mid_snr": -2.042182,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 1.263001,
+    "fb_fre_high_snr": -2.027766,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 1.300019,
+    "fb_fre_very_high_snr": -2.278716,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "denoised": {
+    "snr": 20.8505,
+    "psnr": 46.0262,
+    "ssim": 0.99,
+    "mae": 0.003184,
+    "mse": 2.5698134777485393e-05,
+    "rmse": 0.005033,
+    "eb_wse_very_weak_5_20_ne": 3.17002,
+    "eb_wse_very_weak_5_20_snr": -10.020176,
+    "eb_wse_weak_20_40_ne": 1.620421,
+    "eb_wse_weak_20_40_snr": -4.192383,
+    "eb_wse_medium_40_70_ne": 0.859789,
+    "eb_wse_medium_40_70_snr": 1.312481,
+    "eb_wse_strong_70_100_ne": 0.073771,
+    "eb_wse_strong_70_100_snr": 22.643215,
+    "fb_fre_low_ne": 0.121402,
+    "fb_fre_low_snr": 18.317273,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.068111,
+    "fb_fre_mid_snr": 23.335695,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.093913,
+    "fb_fre_high_snr": 20.569426,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 0.558959,
+    "fb_fre_very_high_snr": 5.138073,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "delta": {
+    "snr": 22.87974,
+    "psnr": 22.879738,
+    "ssim": 0.287199,
+    "mae": -0.022247,
+    "mse": -0.005003,
+    "rmse": -0.065226,
+    "eb_wse_very_weak_5_20_ne": -128.545239,
+    "eb_wse_very_weak_5_20_snr": 32.332384,
+    "eb_wse_weak_20_40_ne": -53.135737,
+    "eb_wse_weak_20_40_snr": 30.551436,
+    "eb_wse_medium_40_70_ne": -15.347753,
+    "eb_wse_medium_40_70_snr": 25.494185,
+    "eb_wse_strong_70_100_ne": -0.138144,
+    "eb_wse_strong_70_100_snr": 9.131659,
+    "fb_fre_low_ne": -1.129652,
+    "fb_fre_low_snr": 20.262446,
+    "fb_fre_mid_ne": -1.196986,
+    "fb_fre_mid_snr": 25.377877,
+    "fb_fre_high_ne": -1.169088,
+    "fb_fre_high_snr": 22.597192,
+    "fb_fre_very_high_ne": -0.74106,
+    "fb_fre_very_high_snr": 7.416789
+  },
+  "input_path": "/data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy",
+  "target_path": "/data/data/T03_seismic_common_receiver.sgy",
+  "input_source": "paired_pseudo_deblended_common_receiver",
+  "inference_time_seconds": 2.005,
+  "num_test_gathers": 12,
+  "snr_min_signal_energy": 1e-08,
+  "input_snr_valid_gathers": 12,
+  "denoised_snr_valid_gathers": 12,
+  "snr_total_gathers": 12,
+  "restored": {
+    "snr": 20.8505,
+    "psnr": 46.0262,
+    "ssim": 0.99,
+    "mae": 0.003184,
+    "mse": 2.5698134777485393e-05,
+    "rmse": 0.005033,
+    "eb_wse_very_weak_5_20_ne": 3.17002,
+    "eb_wse_very_weak_5_20_snr": -10.020176,
+    "eb_wse_weak_20_40_ne": 1.620421,
+    "eb_wse_weak_20_40_snr": -4.192383,
+    "eb_wse_medium_40_70_ne": 0.859789,
+    "eb_wse_medium_40_70_snr": 1.312481,
+    "eb_wse_strong_70_100_ne": 0.073771,
+    "eb_wse_strong_70_100_snr": 22.643215,
+    "fb_fre_low_ne": 0.121402,
+    "fb_fre_low_snr": 18.317273,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.068111,
+    "fb_fre_mid_snr": 23.335695,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.093913,
+    "fb_fre_high_snr": 20.569426,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 0.558959,
+    "fb_fre_very_high_snr": 5.138073,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  }
+}

blending-noise-suppression-avo/models/atten_unet/T03_avo_mod_seed42/model_metadata.json

@@ -0,0 +1,7 @@
+{
+  "task": "blending_noise_suppression",
+  "model": "atten_unet",
+  "variant": "T03_avo_mod",
+  "seed": "42",
+  "num_parameters": 7850000
+}

blending-noise-suppression-avo/models/atten_unet/T03_avo_mod_seed43/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bafeae36cdbb61a22f73bf924ae8b672bb52aa872aa72d1a2fa03f272799c101
+size 86901303

blending-noise-suppression-avo/models/atten_unet/T03_avo_mod_seed43/config.yaml

@@ -0,0 +1,126 @@
+experiment:
+  name: blending_noise_avo_atten_unet_base_T03_avo_mod_seed43_T03_pseudo_deblended_seismic_common_receiver_mod
+  output_dir: /data/data/results/blending_noise_avo/atten_unet
+  seed: 43
+  device: cuda
+data:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  loader:
+    batch_size: 360
+    num_workers: 4
+    pin_memory: true
+preprocess:
+  dt: 0.004
+  normalize_mode: max_abs
+  normalize_scope: shot
+  patch_time: 256
+  patch_trace: 128
+  patch_overlap: 0.5
+  max_gathers: null
+  skip: []
+model:
+  type: atten_unet
+  params:
+    in_channels: 1
+    out_channels: 1
+    base_channels: 32
+    depth: 4
+loss:
+  type: mse
+  params:
+    reduction: mean
+metrics:
+- name: snr
+  params:
+    reduction: per_sample
+- name: psnr
+  params:
+    data_range: 1.0
+    reduction: per_sample
+- name: ssim
+  params:
+    data_range: 2.0
+    window_size: 11
+    sigma: 1.5
+- name: mae
+  params: {}
+- name: mse
+  params: {}
+- name: rmse
+  params:
+    reduction: per_sample
+metrics_filter:
+  snr_min_signal_energy: 1.0e-08
+optim:
+  type: adamw
+  params:
+    lr: 0.0001
+    weight_decay: 1.0e-05
+scheduler:
+  type: cosine
+  params:
+    min_lr: 1.0e-06
+train:
+  epochs: 200
+  grad_clip: 1.0
+  log_step: false
+  log_interval: 10
+  eval_interval: 1
+  ckpt_interval: 20
+  vis_interval: 5
+  resume: null
+log:
+  log_dir: logs
+  plot_interval: 5
+inference:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  checkpoint: /data/data/results/blending_noise_avo/atten_unet/blending_noise_avo_atten_unet_base/checkpoints/best.pt
+  output_dir: /data/data/results/blending_noise_avo/atten_unet/blending_noise_avo_atten_unet_base/inference_integrated_metrics
+  n_viz_gathers: 5
+  device: cuda:0
+  batch_size: 360
+  save_npy: false
+  binned_metrics:
+    enabled: true
+    eb_wse:
+      enabled: true
+      bins:
+      - - 5
+        - 20
+      - - 20
+        - 40
+      - - 40
+        - 70
+      - - 70
+        - 100
+      smooth_sigma: 1.0
+    fb_fre:
+      enabled: true
+      rel_threshold: 0.001
+      band_ratios:
+      - 0.2
+      - 0.3
+      - 0.3
+      - 0.2
+      band_names:
+      - low
+      - mid
+      - high
+      - very_high
+      taper_width: null

blending-noise-suppression-avo/models/atten_unet/T03_avo_mod_seed43/metrics_summary.json

@@ -0,0 +1,170 @@
+{
+  "task": "blending_noise_suppression",
+  "sample_kind": "gather",
+  "sample_axis_name": "common_receiver_gather",
+  "input": {
+    "snr": -2.0292,
+    "psnr": 23.1464,
+    "ssim": 0.7028,
+    "mae": 0.025431,
+    "mse": 0.005028636660426855,
+    "rmse": 0.070259,
+    "eb_wse_very_weak_5_20_ne": 131.715259,
+    "eb_wse_very_weak_5_20_snr": -42.35256,
+    "eb_wse_weak_20_40_ne": 54.756158,
+    "eb_wse_weak_20_40_snr": -34.743819,
+    "eb_wse_medium_40_70_ne": 16.207542,
+    "eb_wse_medium_40_70_snr": -24.181704,
+    "eb_wse_strong_70_100_ne": 0.211915,
+    "eb_wse_strong_70_100_snr": 13.511556,
+    "fb_fre_low_ne": 1.251054,
+    "fb_fre_low_snr": -1.945173,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 1.265097,
+    "fb_fre_mid_snr": -2.042182,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 1.263001,
+    "fb_fre_high_snr": -2.027766,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 1.300019,
+    "fb_fre_very_high_snr": -2.278716,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "denoised": {
+    "snr": 20.0068,
+    "psnr": 45.1825,
+    "ssim": 0.9895,
+    "mae": 0.003372,
+    "mse": 3.1578474590787664e-05,
+    "rmse": 0.005563,
+    "eb_wse_very_weak_5_20_ne": 3.303877,
+    "eb_wse_very_weak_5_20_snr": -10.357475,
+    "eb_wse_weak_20_40_ne": 1.594006,
+    "eb_wse_weak_20_40_snr": -4.041117,
+    "eb_wse_medium_40_70_ne": 0.852541,
+    "eb_wse_medium_40_70_snr": 1.385709,
+    "eb_wse_strong_70_100_ne": 0.084628,
+    "eb_wse_strong_70_100_snr": 21.459306,
+    "fb_fre_low_ne": 0.126728,
+    "fb_fre_low_snr": 17.946621,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.075794,
+    "fb_fre_mid_snr": 22.408565,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.107507,
+    "fb_fre_high_snr": 19.38246,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 0.617994,
+    "fb_fre_very_high_snr": 4.244582,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "delta": {
+    "snr": 22.036089,
+    "psnr": 22.036087,
+    "ssim": 0.286708,
+    "mae": -0.022059,
+    "mse": -0.004997,
+    "rmse": -0.064697,
+    "eb_wse_very_weak_5_20_ne": -128.411382,
+    "eb_wse_very_weak_5_20_snr": 31.995085,
+    "eb_wse_weak_20_40_ne": -53.162152,
+    "eb_wse_weak_20_40_snr": 30.702702,
+    "eb_wse_medium_40_70_ne": -15.355001,
+    "eb_wse_medium_40_70_snr": 25.567413,
+    "eb_wse_strong_70_100_ne": -0.127287,
+    "eb_wse_strong_70_100_snr": 7.94775,
+    "fb_fre_low_ne": -1.124326,
+    "fb_fre_low_snr": 19.891794,
+    "fb_fre_mid_ne": -1.189303,
+    "fb_fre_mid_snr": 24.450747,
+    "fb_fre_high_ne": -1.155494,
+    "fb_fre_high_snr": 21.410226,
+    "fb_fre_very_high_ne": -0.682025,
+    "fb_fre_very_high_snr": 6.523298
+  },
+  "input_path": "/data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy",
+  "target_path": "/data/data/T03_seismic_common_receiver.sgy",
+  "input_source": "paired_pseudo_deblended_common_receiver",
+  "inference_time_seconds": 1.993,
+  "num_test_gathers": 12,
+  "snr_min_signal_energy": 1e-08,
+  "input_snr_valid_gathers": 12,
+  "denoised_snr_valid_gathers": 12,
+  "snr_total_gathers": 12,
+  "restored": {
+    "snr": 20.0068,
+    "psnr": 45.1825,
+    "ssim": 0.9895,
+    "mae": 0.003372,
+    "mse": 3.1578474590787664e-05,
+    "rmse": 0.005563,
+    "eb_wse_very_weak_5_20_ne": 3.303877,
+    "eb_wse_very_weak_5_20_snr": -10.357475,
+    "eb_wse_weak_20_40_ne": 1.594006,
+    "eb_wse_weak_20_40_snr": -4.041117,
+    "eb_wse_medium_40_70_ne": 0.852541,
+    "eb_wse_medium_40_70_snr": 1.385709,
+    "eb_wse_strong_70_100_ne": 0.084628,
+    "eb_wse_strong_70_100_snr": 21.459306,
+    "fb_fre_low_ne": 0.126728,
+    "fb_fre_low_snr": 17.946621,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.075794,
+    "fb_fre_mid_snr": 22.408565,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.107507,
+    "fb_fre_high_snr": 19.38246,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 0.617994,
+    "fb_fre_very_high_snr": 4.244582,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  }
+}

blending-noise-suppression-avo/models/atten_unet/T03_avo_mod_seed43/model_metadata.json

@@ -0,0 +1,7 @@
+{
+  "task": "blending_noise_suppression",
+  "model": "atten_unet",
+  "variant": "T03_avo_mod",
+  "seed": "43",
+  "num_parameters": 7850000
+}

blending-noise-suppression-avo/models/atten_unet/T03_avo_mod_seed44/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:09445449eaf14549a6a34e707295bfbb074427d05d39ab331d779ce49523b1fa
+size 86586444

blending-noise-suppression-avo/models/atten_unet/T03_avo_mod_seed44/config.yaml

@@ -0,0 +1,126 @@
+experiment:
+  name: blending_noise_avo_atten_unet_base_T03_avo_mod_seed44_T03_pseudo_deblended_seismic_common_receiver_mod
+  output_dir: /data/data/results/blending_noise_avo/atten_unet
+  seed: 44
+  device: cuda
+data:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  loader:
+    batch_size: 360
+    num_workers: 4
+    pin_memory: true
+preprocess:
+  dt: 0.004
+  normalize_mode: max_abs
+  normalize_scope: shot
+  patch_time: 256
+  patch_trace: 128
+  patch_overlap: 0.5
+  max_gathers: null
+  skip: []
+model:
+  type: atten_unet
+  params:
+    in_channels: 1
+    out_channels: 1
+    base_channels: 32
+    depth: 4
+loss:
+  type: mse
+  params:
+    reduction: mean
+metrics:
+- name: snr
+  params:
+    reduction: per_sample
+- name: psnr
+  params:
+    data_range: 1.0
+    reduction: per_sample
+- name: ssim
+  params:
+    data_range: 2.0
+    window_size: 11
+    sigma: 1.5
+- name: mae
+  params: {}
+- name: mse
+  params: {}
+- name: rmse
+  params:
+    reduction: per_sample
+metrics_filter:
+  snr_min_signal_energy: 1.0e-08
+optim:
+  type: adamw
+  params:
+    lr: 0.0001
+    weight_decay: 1.0e-05
+scheduler:
+  type: cosine
+  params:
+    min_lr: 1.0e-06
+train:
+  epochs: 200
+  grad_clip: 1.0
+  log_step: false
+  log_interval: 10
+  eval_interval: 1
+  ckpt_interval: 20
+  vis_interval: 5
+  resume: null
+log:
+  log_dir: logs
+  plot_interval: 5
+inference:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  checkpoint: /data/data/results/blending_noise_avo/atten_unet/blending_noise_avo_atten_unet_base/checkpoints/best.pt
+  output_dir: /data/data/results/blending_noise_avo/atten_unet/blending_noise_avo_atten_unet_base/inference_integrated_metrics
+  n_viz_gathers: 5
+  device: cuda:0
+  batch_size: 360
+  save_npy: false
+  binned_metrics:
+    enabled: true
+    eb_wse:
+      enabled: true
+      bins:
+      - - 5
+        - 20
+      - - 20
+        - 40
+      - - 40
+        - 70
+      - - 70
+        - 100
+      smooth_sigma: 1.0
+    fb_fre:
+      enabled: true
+      rel_threshold: 0.001
+      band_ratios:
+      - 0.2
+      - 0.3
+      - 0.3
+      - 0.2
+      band_names:
+      - low
+      - mid
+      - high
+      - very_high
+      taper_width: null

blending-noise-suppression-avo/models/atten_unet/T03_avo_mod_seed44/metrics_summary.json

@@ -0,0 +1,170 @@
+{
+  "task": "blending_noise_suppression",
+  "sample_kind": "gather",
+  "sample_axis_name": "common_receiver_gather",
+  "input": {
+    "snr": -2.0292,
+    "psnr": 23.1464,
+    "ssim": 0.7028,
+    "mae": 0.025431,
+    "mse": 0.005028636660426855,
+    "rmse": 0.070259,
+    "eb_wse_very_weak_5_20_ne": 131.715259,
+    "eb_wse_very_weak_5_20_snr": -42.35256,
+    "eb_wse_weak_20_40_ne": 54.756158,
+    "eb_wse_weak_20_40_snr": -34.743819,
+    "eb_wse_medium_40_70_ne": 16.207542,
+    "eb_wse_medium_40_70_snr": -24.181704,
+    "eb_wse_strong_70_100_ne": 0.211915,
+    "eb_wse_strong_70_100_snr": 13.511556,
+    "fb_fre_low_ne": 1.251054,
+    "fb_fre_low_snr": -1.945173,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 1.265097,
+    "fb_fre_mid_snr": -2.042182,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 1.263001,
+    "fb_fre_high_snr": -2.027766,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 1.300019,
+    "fb_fre_very_high_snr": -2.278716,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "denoised": {
+    "snr": 20.8965,
+    "psnr": 46.0722,
+    "ssim": 0.9906,
+    "mae": 0.003093,
+    "mse": 2.5562150767655112e-05,
+    "rmse": 0.005013,
+    "eb_wse_very_weak_5_20_ne": 2.811298,
+    "eb_wse_very_weak_5_20_snr": -8.977956,
+    "eb_wse_weak_20_40_ne": 1.441547,
+    "eb_wse_weak_20_40_snr": -3.175038,
+    "eb_wse_medium_40_70_ne": 0.841651,
+    "eb_wse_medium_40_70_snr": 1.49807,
+    "eb_wse_strong_70_100_ne": 0.074419,
+    "eb_wse_strong_70_100_snr": 22.56936,
+    "fb_fre_low_ne": 0.120736,
+    "fb_fre_low_snr": 18.367446,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.066902,
+    "fb_fre_mid_snr": 23.49151,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.093189,
+    "fb_fre_high_snr": 20.633187,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 0.577191,
+    "fb_fre_very_high_snr": 4.836806,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "delta": {
+    "snr": 22.925781,
+    "psnr": 22.925779,
+    "ssim": 0.287761,
+    "mae": -0.022337,
+    "mse": -0.005003,
+    "rmse": -0.065246,
+    "eb_wse_very_weak_5_20_ne": -128.903961,
+    "eb_wse_very_weak_5_20_snr": 33.374604,
+    "eb_wse_weak_20_40_ne": -53.314611,
+    "eb_wse_weak_20_40_snr": 31.568781,
+    "eb_wse_medium_40_70_ne": -15.365891,
+    "eb_wse_medium_40_70_snr": 25.679774,
+    "eb_wse_strong_70_100_ne": -0.137496,
+    "eb_wse_strong_70_100_snr": 9.057804,
+    "fb_fre_low_ne": -1.130318,
+    "fb_fre_low_snr": 20.312619,
+    "fb_fre_mid_ne": -1.198195,
+    "fb_fre_mid_snr": 25.533692,
+    "fb_fre_high_ne": -1.169812,
+    "fb_fre_high_snr": 22.660953,
+    "fb_fre_very_high_ne": -0.722828,
+    "fb_fre_very_high_snr": 7.115522
+  },
+  "input_path": "/data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy",
+  "target_path": "/data/data/T03_seismic_common_receiver.sgy",
+  "input_source": "paired_pseudo_deblended_common_receiver",
+  "inference_time_seconds": 2.115,
+  "num_test_gathers": 12,
+  "snr_min_signal_energy": 1e-08,
+  "input_snr_valid_gathers": 12,
+  "denoised_snr_valid_gathers": 12,
+  "snr_total_gathers": 12,
+  "restored": {
+    "snr": 20.8965,
+    "psnr": 46.0722,
+    "ssim": 0.9906,
+    "mae": 0.003093,
+    "mse": 2.5562150767655112e-05,
+    "rmse": 0.005013,
+    "eb_wse_very_weak_5_20_ne": 2.811298,
+    "eb_wse_very_weak_5_20_snr": -8.977956,
+    "eb_wse_weak_20_40_ne": 1.441547,
+    "eb_wse_weak_20_40_snr": -3.175038,
+    "eb_wse_medium_40_70_ne": 0.841651,
+    "eb_wse_medium_40_70_snr": 1.49807,
+    "eb_wse_strong_70_100_ne": 0.074419,
+    "eb_wse_strong_70_100_snr": 22.56936,
+    "fb_fre_low_ne": 0.120736,
+    "fb_fre_low_snr": 18.367446,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.066902,
+    "fb_fre_mid_snr": 23.49151,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.093189,
+    "fb_fre_high_snr": 20.633187,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 0.577191,
+    "fb_fre_very_high_snr": 4.836806,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  }
+}

blending-noise-suppression-avo/models/atten_unet/T03_avo_mod_seed44/model_metadata.json

@@ -0,0 +1,7 @@
+{
+  "task": "blending_noise_suppression",
+  "model": "atten_unet",
+  "variant": "T03_avo_mod",
+  "seed": "44",
+  "num_parameters": 7850000
+}

blending-noise-suppression-avo/models/ddpm/T03_avo_mod_seed42/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5f6491fe9b604f1a5574c368bb8e87771aba464da7bdb20f94eb3b57cca2487d
+size 366434710

blending-noise-suppression-avo/models/ddpm/T03_avo_mod_seed42/config.yaml

@@ -0,0 +1,147 @@
+experiment:
+  name: blending_noise_avo_ddpm_base_T03_avo_mod_seed42_T03_pseudo_deblended_seismic_common_receiver_mod
+  output_dir: /data/data/results/blending_noise_avo/ddpm
+  seed: 42
+  device: cuda
+data:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  loader:
+    batch_size: 100
+    num_workers: 4
+    pin_memory: true
+preprocess:
+  dt: 0.004
+  normalize_mode: max_abs
+  normalize_scope: shot
+  patch_time: 256
+  patch_trace: 128
+  patch_overlap: 0.5
+  max_gathers: null
+  skip: []
+model:
+  type: ddpm
+  params:
+    in_channels: 2
+    out_channels: 1
+    base_channels: 64
+    channel_mults:
+    - 1
+    - 2
+    - 4
+    - 8
+    time_emb_dim: 256
+    num_res_blocks: 1
+    use_attention: true
+loss:
+  type: mse
+  params:
+    reduction: mean
+reconstruction_loss:
+  type: mse
+  params:
+    reduction: mean
+diffusion:
+  num_timesteps: 1000
+  beta_start: 0.0001
+  beta_end: 0.02
+  eval_sample_steps: 10
+  inference_steps: 100
+  use_ddim: true
+  ddim_eta: 0.0
+metrics:
+- name: snr
+  params:
+    reduction: per_sample
+- name: psnr
+  params:
+    data_range: 1.0
+    reduction: per_sample
+- name: ssim
+  params:
+    data_range: 2.0
+    window_size: 11
+    sigma: 1.5
+- name: mae
+  params: {}
+- name: mse
+  params: {}
+- name: rmse
+  params:
+    reduction: per_sample
+metrics_filter:
+  snr_min_signal_energy: 1.0e-08
+optim:
+  type: adamw
+  params:
+    lr: 0.0001
+    weight_decay: 1.0e-05
+scheduler:
+  type: cosine
+  params:
+    min_lr: 1.0e-06
+train:
+  epochs: 200
+  deterministic: false
+  grad_clip: 1.0
+  log_step: false
+  log_interval: 10
+  eval_interval: 5
+  eval_train_metrics: false
+  ckpt_interval: 20
+  vis_interval: 5
+  resume: null
+log:
+  log_dir: logs
+  plot_interval: 5
+inference:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  checkpoint: /data/data/results/blending_noise_avo/ddpm/blending_noise_avo_ddpm_base/checkpoints/best.pt
+  output_dir: /data/data/results/blending_noise_avo/ddpm/blending_noise_avo_ddpm_base/inference_integrated_metrics
+  n_viz_gathers: 5
+  device: cuda:0
+  batch_size: 100
+  save_npy: false
+  binned_metrics:
+    enabled: true
+    eb_wse:
+      enabled: true
+      bins:
+      - - 5
+        - 20
+      - - 20
+        - 40
+      - - 40
+        - 70
+      - - 70
+        - 100
+      smooth_sigma: 1.0
+    fb_fre:
+      enabled: true
+      rel_threshold: 0.001
+      band_ratios:
+      - 0.2
+      - 0.3
+      - 0.3
+      - 0.2
+      band_names:
+      - low
+      - mid
+      - high
+      - very_high
+      taper_width: null

blending-noise-suppression-avo/models/ddpm/T03_avo_mod_seed42/metrics_summary.json

@@ -0,0 +1,166 @@
+{
+  "input": {
+    "snr": -2.0292,
+    "psnr": 23.1464,
+    "ssim": 0.7028,
+    "mae": 0.025431,
+    "mse": 0.005028636660426855,
+    "rmse": 0.070259,
+    "eb_wse_very_weak_5_20_ne": 131.715259,
+    "eb_wse_very_weak_5_20_snr": -42.35256,
+    "eb_wse_weak_20_40_ne": 54.756158,
+    "eb_wse_weak_20_40_snr": -34.743819,
+    "eb_wse_medium_40_70_ne": 16.207542,
+    "eb_wse_medium_40_70_snr": -24.181704,
+    "eb_wse_strong_70_100_ne": 0.211915,
+    "eb_wse_strong_70_100_snr": 13.511556,
+    "fb_fre_low_ne": 1.251054,
+    "fb_fre_low_snr": -1.945173,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 1.265097,
+    "fb_fre_mid_snr": -2.042182,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 1.263001,
+    "fb_fre_high_snr": -2.027766,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 1.300019,
+    "fb_fre_very_high_snr": -2.278716,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "denoised": {
+    "snr": 18.333,
+    "psnr": 43.5086,
+    "ssim": 0.9818,
+    "mae": 0.005126,
+    "mse": 4.498199268709868e-05,
+    "rmse": 0.006692,
+    "eb_wse_very_weak_5_20_ne": 9.626885,
+    "eb_wse_very_weak_5_20_snr": -19.660895,
+    "eb_wse_weak_20_40_ne": 4.097857,
+    "eb_wse_weak_20_40_snr": -12.24849,
+    "eb_wse_medium_40_70_ne": 1.296113,
+    "eb_wse_medium_40_70_snr": -2.251162,
+    "eb_wse_strong_70_100_ne": 0.069136,
+    "eb_wse_strong_70_100_snr": 23.225181,
+    "fb_fre_low_ne": 0.146911,
+    "fb_fre_low_snr": 16.661438,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.076646,
+    "fb_fre_mid_snr": 22.317164,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.12796,
+    "fb_fre_high_snr": 18.000725,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 0.957466,
+    "fb_fre_very_high_snr": 0.613494,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "delta": {
+    "snr": 20.362217,
+    "psnr": 20.362219,
+    "ssim": 0.279046,
+    "mae": -0.020305,
+    "mse": -0.004984,
+    "rmse": -0.063567,
+    "eb_wse_very_weak_5_20_ne": -122.088374,
+    "eb_wse_very_weak_5_20_snr": 22.691665,
+    "eb_wse_weak_20_40_ne": -50.658301,
+    "eb_wse_weak_20_40_snr": 22.495329,
+    "eb_wse_medium_40_70_ne": -14.911429,
+    "eb_wse_medium_40_70_snr": 21.930542,
+    "eb_wse_strong_70_100_ne": -0.142779,
+    "eb_wse_strong_70_100_snr": 9.713625,
+    "fb_fre_low_ne": -1.104143,
+    "fb_fre_low_snr": 18.606611,
+    "fb_fre_mid_ne": -1.188451,
+    "fb_fre_mid_snr": 24.359346,
+    "fb_fre_high_ne": -1.135041,
+    "fb_fre_high_snr": 20.028491,
+    "fb_fre_very_high_ne": -0.342553,
+    "fb_fre_very_high_snr": 2.89221
+  },
+  "input_source": "paired_blended_common_receiver_gathers",
+  "sample_steps": 100,
+  "use_ddim": true,
+  "ddim_eta": 0.0,
+  "inference_time_seconds": 641.727,
+  "input_snr_valid_count": 12,
+  "denoised_snr_valid_count": 12,
+  "snr_total_count": 12,
+  "restored": {
+    "snr": 18.333,
+    "psnr": 43.5086,
+    "ssim": 0.9818,
+    "mae": 0.005126,
+    "mse": 4.498199268709868e-05,
+    "rmse": 0.006692,
+    "eb_wse_very_weak_5_20_ne": 9.626885,
+    "eb_wse_very_weak_5_20_snr": -19.660895,
+    "eb_wse_weak_20_40_ne": 4.097857,
+    "eb_wse_weak_20_40_snr": -12.24849,
+    "eb_wse_medium_40_70_ne": 1.296113,
+    "eb_wse_medium_40_70_snr": -2.251162,
+    "eb_wse_strong_70_100_ne": 0.069136,
+    "eb_wse_strong_70_100_snr": 23.225181,
+    "fb_fre_low_ne": 0.146911,
+    "fb_fre_low_snr": 16.661438,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.076646,
+    "fb_fre_mid_snr": 22.317164,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.12796,
+    "fb_fre_high_snr": 18.000725,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 0.957466,
+    "fb_fre_very_high_snr": 0.613494,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  }
+}

blending-noise-suppression-avo/models/ddpm/T03_avo_mod_seed42/model_metadata.json

@@ -0,0 +1,7 @@
+{
+  "task": "blending_noise_suppression",
+  "model": "ddpm",
+  "variant": "T03_avo_mod",
+  "seed": "42",
+  "num_parameters": 33300000
+}

blending-noise-suppression-avo/models/ddpm/T03_avo_mod_seed43/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:173678c293817901213b52030e8db719a5b2558db25eea0f0b284b3edf9de80e
+size 366780801

blending-noise-suppression-avo/models/ddpm/T03_avo_mod_seed43/config.yaml

@@ -0,0 +1,147 @@
+experiment:
+  name: blending_noise_avo_ddpm_base_T03_avo_mod_seed43_T03_pseudo_deblended_seismic_common_receiver_mod
+  output_dir: /data/data/results/blending_noise_avo/ddpm
+  seed: 43
+  device: cuda
+data:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  loader:
+    batch_size: 100
+    num_workers: 4
+    pin_memory: true
+preprocess:
+  dt: 0.004
+  normalize_mode: max_abs
+  normalize_scope: shot
+  patch_time: 256
+  patch_trace: 128
+  patch_overlap: 0.5
+  max_gathers: null
+  skip: []
+model:
+  type: ddpm
+  params:
+    in_channels: 2
+    out_channels: 1
+    base_channels: 64
+    channel_mults:
+    - 1
+    - 2
+    - 4
+    - 8
+    time_emb_dim: 256
+    num_res_blocks: 1
+    use_attention: true
+loss:
+  type: mse
+  params:
+    reduction: mean
+reconstruction_loss:
+  type: mse
+  params:
+    reduction: mean
+diffusion:
+  num_timesteps: 1000
+  beta_start: 0.0001
+  beta_end: 0.02
+  eval_sample_steps: 10
+  inference_steps: 100
+  use_ddim: true
+  ddim_eta: 0.0
+metrics:
+- name: snr
+  params:
+    reduction: per_sample
+- name: psnr
+  params:
+    data_range: 1.0
+    reduction: per_sample
+- name: ssim
+  params:
+    data_range: 2.0
+    window_size: 11
+    sigma: 1.5
+- name: mae
+  params: {}
+- name: mse
+  params: {}
+- name: rmse
+  params:
+    reduction: per_sample
+metrics_filter:
+  snr_min_signal_energy: 1.0e-08
+optim:
+  type: adamw
+  params:
+    lr: 0.0001
+    weight_decay: 1.0e-05
+scheduler:
+  type: cosine
+  params:
+    min_lr: 1.0e-06
+train:
+  epochs: 200
+  deterministic: false
+  grad_clip: 1.0
+  log_step: false
+  log_interval: 10
+  eval_interval: 5
+  eval_train_metrics: false
+  ckpt_interval: 20
+  vis_interval: 5
+  resume: null
+log:
+  log_dir: logs
+  plot_interval: 5
+inference:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  checkpoint: /data/data/results/blending_noise_avo/ddpm/blending_noise_avo_ddpm_base/checkpoints/best.pt
+  output_dir: /data/data/results/blending_noise_avo/ddpm/blending_noise_avo_ddpm_base/inference_integrated_metrics
+  n_viz_gathers: 5
+  device: cuda:0
+  batch_size: 100
+  save_npy: false
+  binned_metrics:
+    enabled: true
+    eb_wse:
+      enabled: true
+      bins:
+      - - 5
+        - 20
+      - - 20
+        - 40
+      - - 40
+        - 70
+      - - 70
+        - 100
+      smooth_sigma: 1.0
+    fb_fre:
+      enabled: true
+      rel_threshold: 0.001
+      band_ratios:
+      - 0.2
+      - 0.3
+      - 0.3
+      - 0.2
+      band_names:
+      - low
+      - mid
+      - high
+      - very_high
+      taper_width: null

blending-noise-suppression-avo/models/ddpm/T03_avo_mod_seed43/metrics_summary.json

@@ -0,0 +1,166 @@
+{
+  "input": {
+    "snr": -2.0292,
+    "psnr": 23.1464,
+    "ssim": 0.7028,
+    "mae": 0.025431,
+    "mse": 0.005028636660426855,
+    "rmse": 0.070259,
+    "eb_wse_very_weak_5_20_ne": 131.715259,
+    "eb_wse_very_weak_5_20_snr": -42.35256,
+    "eb_wse_weak_20_40_ne": 54.756158,
+    "eb_wse_weak_20_40_snr": -34.743819,
+    "eb_wse_medium_40_70_ne": 16.207542,
+    "eb_wse_medium_40_70_snr": -24.181704,
+    "eb_wse_strong_70_100_ne": 0.211915,
+    "eb_wse_strong_70_100_snr": 13.511556,
+    "fb_fre_low_ne": 1.251054,
+    "fb_fre_low_snr": -1.945173,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 1.265097,
+    "fb_fre_mid_snr": -2.042182,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 1.263001,
+    "fb_fre_high_snr": -2.027766,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 1.300019,
+    "fb_fre_very_high_snr": -2.278716,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "denoised": {
+    "snr": 16.8332,
+    "psnr": 42.0089,
+    "ssim": 0.9764,
+    "mae": 0.005863,
+    "mse": 6.32589653832838e-05,
+    "rmse": 0.007944,
+    "eb_wse_very_weak_5_20_ne": 12.618474,
+    "eb_wse_very_weak_5_20_snr": -21.930134,
+    "eb_wse_weak_20_40_ne": 5.191117,
+    "eb_wse_weak_20_40_snr": -14.241539,
+    "eb_wse_medium_40_70_ne": 1.490873,
+    "eb_wse_medium_40_70_snr": -3.457861,
+    "eb_wse_strong_70_100_ne": 0.078182,
+    "eb_wse_strong_70_100_snr": 22.181513,
+    "fb_fre_low_ne": 0.167431,
+    "fb_fre_low_snr": 15.528914,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.087768,
+    "fb_fre_mid_snr": 21.156729,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.156536,
+    "fb_fre_high_snr": 16.332116,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 1.191098,
+    "fb_fre_very_high_snr": -1.18615,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "delta": {
+    "snr": 18.862478,
+    "psnr": 18.862478,
+    "ssim": 0.27358,
+    "mae": -0.019568,
+    "mse": -0.004965,
+    "rmse": -0.062315,
+    "eb_wse_very_weak_5_20_ne": -119.096785,
+    "eb_wse_very_weak_5_20_snr": 20.422426,
+    "eb_wse_weak_20_40_ne": -49.565041,
+    "eb_wse_weak_20_40_snr": 20.50228,
+    "eb_wse_medium_40_70_ne": -14.716669,
+    "eb_wse_medium_40_70_snr": 20.723843,
+    "eb_wse_strong_70_100_ne": -0.133733,
+    "eb_wse_strong_70_100_snr": 8.669957,
+    "fb_fre_low_ne": -1.083623,
+    "fb_fre_low_snr": 17.474087,
+    "fb_fre_mid_ne": -1.177329,
+    "fb_fre_mid_snr": 23.198911,
+    "fb_fre_high_ne": -1.106465,
+    "fb_fre_high_snr": 18.359882,
+    "fb_fre_very_high_ne": -0.108921,
+    "fb_fre_very_high_snr": 1.092566
+  },
+  "input_source": "paired_blended_common_receiver_gathers",
+  "sample_steps": 100,
+  "use_ddim": true,
+  "ddim_eta": 0.0,
+  "inference_time_seconds": 642.721,
+  "input_snr_valid_count": 12,
+  "denoised_snr_valid_count": 12,
+  "snr_total_count": 12,
+  "restored": {
+    "snr": 16.8332,
+    "psnr": 42.0089,
+    "ssim": 0.9764,
+    "mae": 0.005863,
+    "mse": 6.32589653832838e-05,
+    "rmse": 0.007944,
+    "eb_wse_very_weak_5_20_ne": 12.618474,
+    "eb_wse_very_weak_5_20_snr": -21.930134,
+    "eb_wse_weak_20_40_ne": 5.191117,
+    "eb_wse_weak_20_40_snr": -14.241539,
+    "eb_wse_medium_40_70_ne": 1.490873,
+    "eb_wse_medium_40_70_snr": -3.457861,
+    "eb_wse_strong_70_100_ne": 0.078182,
+    "eb_wse_strong_70_100_snr": 22.181513,
+    "fb_fre_low_ne": 0.167431,
+    "fb_fre_low_snr": 15.528914,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.087768,
+    "fb_fre_mid_snr": 21.156729,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.156536,
+    "fb_fre_high_snr": 16.332116,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 1.191098,
+    "fb_fre_very_high_snr": -1.18615,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  }
+}

blending-noise-suppression-avo/models/ddpm/T03_avo_mod_seed43/model_metadata.json

@@ -0,0 +1,7 @@
+{
+  "task": "blending_noise_suppression",
+  "model": "ddpm",
+  "variant": "T03_avo_mod",
+  "seed": "43",
+  "num_parameters": 33300000
+}

blending-noise-suppression-avo/models/ddpm/T03_avo_mod_seed44/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bd349058c63f42b5c0d6b5c340581e28c2f2c29a5142aaa25c77f9219977ab95
+size 366067781

blending-noise-suppression-avo/models/ddpm/T03_avo_mod_seed44/config.yaml

@@ -0,0 +1,147 @@
+experiment:
+  name: blending_noise_avo_ddpm_base_T03_avo_mod_seed44_T03_pseudo_deblended_seismic_common_receiver_mod
+  output_dir: /data/data/results/blending_noise_avo/ddpm
+  seed: 44
+  device: cuda
+data:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  loader:
+    batch_size: 100
+    num_workers: 4
+    pin_memory: true
+preprocess:
+  dt: 0.004
+  normalize_mode: max_abs
+  normalize_scope: shot
+  patch_time: 256
+  patch_trace: 128
+  patch_overlap: 0.5
+  max_gathers: null
+  skip: []
+model:
+  type: ddpm
+  params:
+    in_channels: 2
+    out_channels: 1
+    base_channels: 64
+    channel_mults:
+    - 1
+    - 2
+    - 4
+    - 8
+    time_emb_dim: 256
+    num_res_blocks: 1
+    use_attention: true
+loss:
+  type: mse
+  params:
+    reduction: mean
+reconstruction_loss:
+  type: mse
+  params:
+    reduction: mean
+diffusion:
+  num_timesteps: 1000
+  beta_start: 0.0001
+  beta_end: 0.02
+  eval_sample_steps: 10
+  inference_steps: 100
+  use_ddim: true
+  ddim_eta: 0.0
+metrics:
+- name: snr
+  params:
+    reduction: per_sample
+- name: psnr
+  params:
+    data_range: 1.0
+    reduction: per_sample
+- name: ssim
+  params:
+    data_range: 2.0
+    window_size: 11
+    sigma: 1.5
+- name: mae
+  params: {}
+- name: mse
+  params: {}
+- name: rmse
+  params:
+    reduction: per_sample
+metrics_filter:
+  snr_min_signal_energy: 1.0e-08
+optim:
+  type: adamw
+  params:
+    lr: 0.0001
+    weight_decay: 1.0e-05
+scheduler:
+  type: cosine
+  params:
+    min_lr: 1.0e-06
+train:
+  epochs: 200
+  deterministic: false
+  grad_clip: 1.0
+  log_step: false
+  log_interval: 10
+  eval_interval: 5
+  eval_train_metrics: false
+  ckpt_interval: 20
+  vis_interval: 5
+  resume: null
+log:
+  log_dir: logs
+  plot_interval: 5
+inference:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  checkpoint: /data/data/results/blending_noise_avo/ddpm/blending_noise_avo_ddpm_base/checkpoints/best.pt
+  output_dir: /data/data/results/blending_noise_avo/ddpm/blending_noise_avo_ddpm_base/inference_integrated_metrics
+  n_viz_gathers: 5
+  device: cuda:0
+  batch_size: 100
+  save_npy: false
+  binned_metrics:
+    enabled: true
+    eb_wse:
+      enabled: true
+      bins:
+      - - 5
+        - 20
+      - - 20
+        - 40
+      - - 40
+        - 70
+      - - 70
+        - 100
+      smooth_sigma: 1.0
+    fb_fre:
+      enabled: true
+      rel_threshold: 0.001
+      band_ratios:
+      - 0.2
+      - 0.3
+      - 0.3
+      - 0.2
+      band_names:
+      - low
+      - mid
+      - high
+      - very_high
+      taper_width: null

blending-noise-suppression-avo/models/ddpm/T03_avo_mod_seed44/metrics_summary.json

@@ -0,0 +1,166 @@
+{
+  "input": {
+    "snr": -2.0292,
+    "psnr": 23.1464,
+    "ssim": 0.7028,
+    "mae": 0.025431,
+    "mse": 0.005028636660426855,
+    "rmse": 0.070259,
+    "eb_wse_very_weak_5_20_ne": 131.715259,
+    "eb_wse_very_weak_5_20_snr": -42.35256,
+    "eb_wse_weak_20_40_ne": 54.756158,
+    "eb_wse_weak_20_40_snr": -34.743819,
+    "eb_wse_medium_40_70_ne": 16.207542,
+    "eb_wse_medium_40_70_snr": -24.181704,
+    "eb_wse_strong_70_100_ne": 0.211915,
+    "eb_wse_strong_70_100_snr": 13.511556,
+    "fb_fre_low_ne": 1.251054,
+    "fb_fre_low_snr": -1.945173,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 1.265097,
+    "fb_fre_mid_snr": -2.042182,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 1.263001,
+    "fb_fre_high_snr": -2.027766,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 1.300019,
+    "fb_fre_very_high_snr": -2.278716,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "denoised": {
+    "snr": 17.6717,
+    "psnr": 42.8474,
+    "ssim": 0.9799,
+    "mae": 0.005503,
+    "mse": 5.229588350630365e-05,
+    "rmse": 0.007218,
+    "eb_wse_very_weak_5_20_ne": 10.769131,
+    "eb_wse_very_weak_5_20_snr": -20.614318,
+    "eb_wse_weak_20_40_ne": 4.575344,
+    "eb_wse_weak_20_40_snr": -13.202645,
+    "eb_wse_medium_40_70_ne": 1.369881,
+    "eb_wse_medium_40_70_snr": -2.727558,
+    "eb_wse_strong_70_100_ne": 0.073081,
+    "eb_wse_strong_70_100_snr": 22.745092,
+    "fb_fre_low_ne": 0.152167,
+    "fb_fre_low_snr": 16.356025,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.080461,
+    "fb_fre_mid_snr": 21.898498,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.140298,
+    "fb_fre_high_snr": 17.224534,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 1.062044,
+    "fb_fre_very_high_snr": -0.261198,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "delta": {
+    "snr": 19.700962,
+    "psnr": 19.700964,
+    "ssim": 0.277076,
+    "mae": -0.019928,
+    "mse": -0.004976,
+    "rmse": -0.063041,
+    "eb_wse_very_weak_5_20_ne": -120.946128,
+    "eb_wse_very_weak_5_20_snr": 21.738242,
+    "eb_wse_weak_20_40_ne": -50.180814,
+    "eb_wse_weak_20_40_snr": 21.541174,
+    "eb_wse_medium_40_70_ne": -14.837661,
+    "eb_wse_medium_40_70_snr": 21.454146,
+    "eb_wse_strong_70_100_ne": -0.138834,
+    "eb_wse_strong_70_100_snr": 9.233536,
+    "fb_fre_low_ne": -1.098887,
+    "fb_fre_low_snr": 18.301198,
+    "fb_fre_mid_ne": -1.184636,
+    "fb_fre_mid_snr": 23.94068,
+    "fb_fre_high_ne": -1.122703,
+    "fb_fre_high_snr": 19.2523,
+    "fb_fre_very_high_ne": -0.237975,
+    "fb_fre_very_high_snr": 2.017518
+  },
+  "input_source": "paired_blended_common_receiver_gathers",
+  "sample_steps": 100,
+  "use_ddim": true,
+  "ddim_eta": 0.0,
+  "inference_time_seconds": 642.466,
+  "input_snr_valid_count": 12,
+  "denoised_snr_valid_count": 12,
+  "snr_total_count": 12,
+  "restored": {
+    "snr": 17.6717,
+    "psnr": 42.8474,
+    "ssim": 0.9799,
+    "mae": 0.005503,
+    "mse": 5.229588350630365e-05,
+    "rmse": 0.007218,
+    "eb_wse_very_weak_5_20_ne": 10.769131,
+    "eb_wse_very_weak_5_20_snr": -20.614318,
+    "eb_wse_weak_20_40_ne": 4.575344,
+    "eb_wse_weak_20_40_snr": -13.202645,
+    "eb_wse_medium_40_70_ne": 1.369881,
+    "eb_wse_medium_40_70_snr": -2.727558,
+    "eb_wse_strong_70_100_ne": 0.073081,
+    "eb_wse_strong_70_100_snr": 22.745092,
+    "fb_fre_low_ne": 0.152167,
+    "fb_fre_low_snr": 16.356025,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.080461,
+    "fb_fre_mid_snr": 21.898498,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.140298,
+    "fb_fre_high_snr": 17.224534,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 1.062044,
+    "fb_fre_very_high_snr": -0.261198,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  }
+}

blending-noise-suppression-avo/models/ddpm/T03_avo_mod_seed44/model_metadata.json

@@ -0,0 +1,7 @@
+{
+  "task": "blending_noise_suppression",
+  "model": "ddpm",
+  "variant": "T03_avo_mod",
+  "seed": "44",
+  "num_parameters": 33300000
+}

blending-noise-suppression-avo/models/dncnn/T03_avo_mod_seed42/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8896f43557247df99f77f358e624c42f8ae7850f443b77c206c880dbe59450df
+size 6247354

blending-noise-suppression-avo/models/dncnn/T03_avo_mod_seed42/config.yaml

@@ -0,0 +1,127 @@
+experiment:
+  name: blending_noise_avo_dncnn_base_T03_avo_mod_seed42_T03_pseudo_deblended_seismic_common_receiver_mod
+  output_dir: /data/data/results/blending_noise_avo/dncnn
+  seed: 42
+  device: cuda
+data:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  loader:
+    batch_size: 142
+    num_workers: 4
+    pin_memory: true
+preprocess:
+  dt: 0.004
+  normalize_mode: max_abs
+  normalize_scope: shot
+  patch_time: 256
+  patch_trace: 128
+  patch_overlap: 0.5
+  max_gathers: null
+  skip: []
+model:
+  type: dncnn
+  params:
+    in_channels: 1
+    out_channels: 1
+    depth: 17
+    base_channels: 64
+    kernel_size: 3
+loss:
+  type: mse
+  params:
+    reduction: mean
+metrics:
+- name: snr
+  params:
+    reduction: per_sample
+- name: psnr
+  params:
+    data_range: 1.0
+    reduction: per_sample
+- name: ssim
+  params:
+    data_range: 2.0
+    window_size: 11
+    sigma: 1.5
+- name: mae
+  params: {}
+- name: mse
+  params: {}
+- name: rmse
+  params:
+    reduction: per_sample
+metrics_filter:
+  snr_min_signal_energy: 1.0e-08
+optim:
+  type: adamw
+  params:
+    lr: 0.0001
+    weight_decay: 1.0e-05
+scheduler:
+  type: cosine
+  params:
+    min_lr: 1.0e-06
+train:
+  epochs: 200
+  grad_clip: 1.0
+  log_step: false
+  log_interval: 10
+  eval_interval: 1
+  ckpt_interval: 20
+  vis_interval: 5
+  resume: null
+log:
+  log_dir: logs
+  plot_interval: 5
+inference:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  checkpoint: /data/data/results/blending_noise_avo/dncnn/blending_noise_avo_dncnn_base/checkpoints/best.pt
+  output_dir: /data/data/results/blending_noise_avo/dncnn/blending_noise_avo_dncnn_base/inference_integrated_metrics
+  n_viz_gathers: 5
+  device: cuda:0
+  batch_size: 142
+  save_npy: false
+  binned_metrics:
+    enabled: true
+    eb_wse:
+      enabled: true
+      bins:
+      - - 5
+        - 20
+      - - 20
+        - 40
+      - - 40
+        - 70
+      - - 70
+        - 100
+      smooth_sigma: 1.0
+    fb_fre:
+      enabled: true
+      rel_threshold: 0.001
+      band_ratios:
+      - 0.2
+      - 0.3
+      - 0.3
+      - 0.2
+      band_names:
+      - low
+      - mid
+      - high
+      - very_high
+      taper_width: null

blending-noise-suppression-avo/models/dncnn/T03_avo_mod_seed42/metrics_summary.json

@@ -0,0 +1,170 @@
+{
+  "task": "blending_noise_suppression",
+  "sample_kind": "gather",
+  "sample_axis_name": "common_receiver_gather",
+  "input": {
+    "snr": -2.0292,
+    "psnr": 23.1464,
+    "ssim": 0.7028,
+    "mae": 0.025431,
+    "mse": 0.005028636660426855,
+    "rmse": 0.070259,
+    "eb_wse_very_weak_5_20_ne": 131.715259,
+    "eb_wse_very_weak_5_20_snr": -42.35256,
+    "eb_wse_weak_20_40_ne": 54.756158,
+    "eb_wse_weak_20_40_snr": -34.743819,
+    "eb_wse_medium_40_70_ne": 16.207542,
+    "eb_wse_medium_40_70_snr": -24.181704,
+    "eb_wse_strong_70_100_ne": 0.211915,
+    "eb_wse_strong_70_100_snr": 13.511556,
+    "fb_fre_low_ne": 1.251054,
+    "fb_fre_low_snr": -1.945173,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 1.265097,
+    "fb_fre_mid_snr": -2.042182,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 1.263001,
+    "fb_fre_high_snr": -2.027766,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 1.300019,
+    "fb_fre_very_high_snr": -2.278716,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "denoised": {
+    "snr": 21.2285,
+    "psnr": 46.4041,
+    "ssim": 0.9877,
+    "mae": 0.003178,
+    "mse": 2.389767178101465e-05,
+    "rmse": 0.004836,
+    "eb_wse_very_weak_5_20_ne": 6.77007,
+    "eb_wse_very_weak_5_20_snr": -16.550956,
+    "eb_wse_weak_20_40_ne": 2.987703,
+    "eb_wse_weak_20_40_snr": -9.461093,
+    "eb_wse_medium_40_70_ne": 1.035339,
+    "eb_wse_medium_40_70_snr": -0.280796,
+    "eb_wse_strong_70_100_ne": 0.046459,
+    "eb_wse_strong_70_100_snr": 26.659822,
+    "fb_fre_low_ne": 0.122446,
+    "fb_fre_low_snr": 18.245846,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.067197,
+    "fb_fre_mid_snr": 23.454593,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.082285,
+    "fb_fre_high_snr": 21.70725,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 0.465813,
+    "fb_fre_very_high_snr": 6.6889,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "delta": {
+    "snr": 23.257719,
+    "psnr": 23.257719,
+    "ssim": 0.28488,
+    "mae": -0.022253,
+    "mse": -0.005005,
+    "rmse": -0.065423,
+    "eb_wse_very_weak_5_20_ne": -124.945189,
+    "eb_wse_very_weak_5_20_snr": 25.801604,
+    "eb_wse_weak_20_40_ne": -51.768455,
+    "eb_wse_weak_20_40_snr": 25.282726,
+    "eb_wse_medium_40_70_ne": -15.172203,
+    "eb_wse_medium_40_70_snr": 23.900908,
+    "eb_wse_strong_70_100_ne": -0.165456,
+    "eb_wse_strong_70_100_snr": 13.148266,
+    "fb_fre_low_ne": -1.128608,
+    "fb_fre_low_snr": 20.191019,
+    "fb_fre_mid_ne": -1.1979,
+    "fb_fre_mid_snr": 25.496775,
+    "fb_fre_high_ne": -1.180716,
+    "fb_fre_high_snr": 23.735016,
+    "fb_fre_very_high_ne": -0.834206,
+    "fb_fre_very_high_snr": 8.967616
+  },
+  "input_path": "/data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy",
+  "target_path": "/data/data/T03_seismic_common_receiver.sgy",
+  "input_source": "paired_pseudo_deblended_common_receiver",
+  "inference_time_seconds": 3.428,
+  "num_test_gathers": 12,
+  "snr_min_signal_energy": 1e-08,
+  "input_snr_valid_gathers": 12,
+  "denoised_snr_valid_gathers": 12,
+  "snr_total_gathers": 12,
+  "restored": {
+    "snr": 21.2285,
+    "psnr": 46.4041,
+    "ssim": 0.9877,
+    "mae": 0.003178,
+    "mse": 2.389767178101465e-05,
+    "rmse": 0.004836,
+    "eb_wse_very_weak_5_20_ne": 6.77007,
+    "eb_wse_very_weak_5_20_snr": -16.550956,
+    "eb_wse_weak_20_40_ne": 2.987703,
+    "eb_wse_weak_20_40_snr": -9.461093,
+    "eb_wse_medium_40_70_ne": 1.035339,
+    "eb_wse_medium_40_70_snr": -0.280796,
+    "eb_wse_strong_70_100_ne": 0.046459,
+    "eb_wse_strong_70_100_snr": 26.659822,
+    "fb_fre_low_ne": 0.122446,
+    "fb_fre_low_snr": 18.245846,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.067197,
+    "fb_fre_mid_snr": 23.454593,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.082285,
+    "fb_fre_high_snr": 21.70725,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 0.465813,
+    "fb_fre_very_high_snr": 6.6889,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  }
+}

blending-noise-suppression-avo/models/dncnn/T03_avo_mod_seed42/model_metadata.json

@@ -0,0 +1,7 @@
+{
+  "task": "blending_noise_suppression",
+  "model": "dncnn",
+  "variant": "T03_avo_mod",
+  "seed": "42",
+  "num_parameters": 560000
+}

blending-noise-suppression-avo/models/dncnn/T03_avo_mod_seed43/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d96184cd2a53aff15fc559153c9d8c8482070a95ad6266aea575330b3c2a0e72
+size 6250094

blending-noise-suppression-avo/models/dncnn/T03_avo_mod_seed43/config.yaml

@@ -0,0 +1,127 @@
+experiment:
+  name: blending_noise_avo_dncnn_base_T03_avo_mod_seed43_T03_pseudo_deblended_seismic_common_receiver_mod
+  output_dir: /data/data/results/blending_noise_avo/dncnn
+  seed: 43
+  device: cuda
+data:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  loader:
+    batch_size: 142
+    num_workers: 4
+    pin_memory: true
+preprocess:
+  dt: 0.004
+  normalize_mode: max_abs
+  normalize_scope: shot
+  patch_time: 256
+  patch_trace: 128
+  patch_overlap: 0.5
+  max_gathers: null
+  skip: []
+model:
+  type: dncnn
+  params:
+    in_channels: 1
+    out_channels: 1
+    depth: 17
+    base_channels: 64
+    kernel_size: 3
+loss:
+  type: mse
+  params:
+    reduction: mean
+metrics:
+- name: snr
+  params:
+    reduction: per_sample
+- name: psnr
+  params:
+    data_range: 1.0
+    reduction: per_sample
+- name: ssim
+  params:
+    data_range: 2.0
+    window_size: 11
+    sigma: 1.5
+- name: mae
+  params: {}
+- name: mse
+  params: {}
+- name: rmse
+  params:
+    reduction: per_sample
+metrics_filter:
+  snr_min_signal_energy: 1.0e-08
+optim:
+  type: adamw
+  params:
+    lr: 0.0001
+    weight_decay: 1.0e-05
+scheduler:
+  type: cosine
+  params:
+    min_lr: 1.0e-06
+train:
+  epochs: 200
+  grad_clip: 1.0
+  log_step: false
+  log_interval: 10
+  eval_interval: 1
+  ckpt_interval: 20
+  vis_interval: 5
+  resume: null
+log:
+  log_dir: logs
+  plot_interval: 5
+inference:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  checkpoint: /data/data/results/blending_noise_avo/dncnn/blending_noise_avo_dncnn_base/checkpoints/best.pt
+  output_dir: /data/data/results/blending_noise_avo/dncnn/blending_noise_avo_dncnn_base/inference_integrated_metrics
+  n_viz_gathers: 5
+  device: cuda:0
+  batch_size: 142
+  save_npy: false
+  binned_metrics:
+    enabled: true
+    eb_wse:
+      enabled: true
+      bins:
+      - - 5
+        - 20
+      - - 20
+        - 40
+      - - 40
+        - 70
+      - - 70
+        - 100
+      smooth_sigma: 1.0
+    fb_fre:
+      enabled: true
+      rel_threshold: 0.001
+      band_ratios:
+      - 0.2
+      - 0.3
+      - 0.3
+      - 0.2
+      band_names:
+      - low
+      - mid
+      - high
+      - very_high
+      taper_width: null

blending-noise-suppression-avo/models/dncnn/T03_avo_mod_seed43/metrics_summary.json

@@ -0,0 +1,170 @@
+{
+  "task": "blending_noise_suppression",
+  "sample_kind": "gather",
+  "sample_axis_name": "common_receiver_gather",
+  "input": {
+    "snr": -2.0292,
+    "psnr": 23.1464,
+    "ssim": 0.7028,
+    "mae": 0.025431,
+    "mse": 0.005028636660426855,
+    "rmse": 0.070259,
+    "eb_wse_very_weak_5_20_ne": 131.715259,
+    "eb_wse_very_weak_5_20_snr": -42.35256,
+    "eb_wse_weak_20_40_ne": 54.756158,
+    "eb_wse_weak_20_40_snr": -34.743819,
+    "eb_wse_medium_40_70_ne": 16.207542,
+    "eb_wse_medium_40_70_snr": -24.181704,
+    "eb_wse_strong_70_100_ne": 0.211915,
+    "eb_wse_strong_70_100_snr": 13.511556,
+    "fb_fre_low_ne": 1.251054,
+    "fb_fre_low_snr": -1.945173,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 1.265097,
+    "fb_fre_mid_snr": -2.042182,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 1.263001,
+    "fb_fre_high_snr": -2.027766,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 1.300019,
+    "fb_fre_very_high_snr": -2.278716,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "denoised": {
+    "snr": 21.1696,
+    "psnr": 46.3453,
+    "ssim": 0.9883,
+    "mae": 0.003199,
+    "mse": 2.4136228603310883e-05,
+    "rmse": 0.004864,
+    "eb_wse_very_weak_5_20_ne": 6.840952,
+    "eb_wse_very_weak_5_20_snr": -16.648278,
+    "eb_wse_weak_20_40_ne": 3.076843,
+    "eb_wse_weak_20_40_snr": -9.72361,
+    "eb_wse_medium_40_70_ne": 1.035419,
+    "eb_wse_medium_40_70_snr": -0.284788,
+    "eb_wse_strong_70_100_ne": 0.046632,
+    "eb_wse_strong_70_100_snr": 26.627387,
+    "fb_fre_low_ne": 0.123973,
+    "fb_fre_low_snr": 18.13707,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.067424,
+    "fb_fre_mid_snr": 23.424608,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.085103,
+    "fb_fre_high_snr": 21.416477,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 0.480136,
+    "fb_fre_very_high_snr": 6.433718,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  },
+  "delta": {
+    "snr": 23.198883,
+    "psnr": 23.198877,
+    "ssim": 0.285517,
+    "mae": -0.022231,
+    "mse": -0.005005,
+    "rmse": -0.065395,
+    "eb_wse_very_weak_5_20_ne": -124.874307,
+    "eb_wse_very_weak_5_20_snr": 25.704282,
+    "eb_wse_weak_20_40_ne": -51.679315,
+    "eb_wse_weak_20_40_snr": 25.020209,
+    "eb_wse_medium_40_70_ne": -15.172123,
+    "eb_wse_medium_40_70_snr": 23.896916,
+    "eb_wse_strong_70_100_ne": -0.165283,
+    "eb_wse_strong_70_100_snr": 13.115831,
+    "fb_fre_low_ne": -1.127081,
+    "fb_fre_low_snr": 20.082243,
+    "fb_fre_mid_ne": -1.197673,
+    "fb_fre_mid_snr": 25.46679,
+    "fb_fre_high_ne": -1.177898,
+    "fb_fre_high_snr": 23.444243,
+    "fb_fre_very_high_ne": -0.819883,
+    "fb_fre_very_high_snr": 8.712434
+  },
+  "input_path": "/data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy",
+  "target_path": "/data/data/T03_seismic_common_receiver.sgy",
+  "input_source": "paired_pseudo_deblended_common_receiver",
+  "inference_time_seconds": 3.391,
+  "num_test_gathers": 12,
+  "snr_min_signal_energy": 1e-08,
+  "input_snr_valid_gathers": 12,
+  "denoised_snr_valid_gathers": 12,
+  "snr_total_gathers": 12,
+  "restored": {
+    "snr": 21.1696,
+    "psnr": 46.3453,
+    "ssim": 0.9883,
+    "mae": 0.003199,
+    "mse": 2.4136228603310883e-05,
+    "rmse": 0.004864,
+    "eb_wse_very_weak_5_20_ne": 6.840952,
+    "eb_wse_very_weak_5_20_snr": -16.648278,
+    "eb_wse_weak_20_40_ne": 3.076843,
+    "eb_wse_weak_20_40_snr": -9.72361,
+    "eb_wse_medium_40_70_ne": 1.035419,
+    "eb_wse_medium_40_70_snr": -0.284788,
+    "eb_wse_strong_70_100_ne": 0.046632,
+    "eb_wse_strong_70_100_snr": 26.627387,
+    "fb_fre_low_ne": 0.123973,
+    "fb_fre_low_snr": 18.13707,
+    "fb_fre_low_energy_ratio": 0.125627,
+    "fb_fre_low_frequency_range_hz": [
+      6.5,
+      23.2
+    ],
+    "fb_fre_mid_ne": 0.067424,
+    "fb_fre_mid_snr": 23.424608,
+    "fb_fre_mid_energy_ratio": 0.637894,
+    "fb_fre_mid_frequency_range_hz": [
+      23.2,
+      48.25
+    ],
+    "fb_fre_high_ne": 0.085103,
+    "fb_fre_high_snr": 21.416477,
+    "fb_fre_high_energy_ratio": 0.140025,
+    "fb_fre_high_frequency_range_hz": [
+      48.25,
+      73.3
+    ],
+    "fb_fre_very_high_ne": 0.480136,
+    "fb_fre_very_high_snr": 6.433718,
+    "fb_fre_very_high_energy_ratio": 0.001545,
+    "fb_fre_very_high_frequency_range_hz": [
+      73.3,
+      90.0
+    ]
+  }
+}

blending-noise-suppression-avo/models/dncnn/T03_avo_mod_seed43/model_metadata.json

@@ -0,0 +1,7 @@
+{
+  "task": "blending_noise_suppression",
+  "model": "dncnn",
+  "variant": "T03_avo_mod",
+  "seed": "43",
+  "num_parameters": 560000
+}

blending-noise-suppression-avo/models/dncnn/T03_avo_mod_seed44/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:31e0397eb56aad5d64bb3a97bcda9b7ed3e534086c5b798603845cd8d34a471f
+size 6244832

blending-noise-suppression-avo/models/dncnn/T03_avo_mod_seed44/config.yaml

@@ -0,0 +1,127 @@
+experiment:
+  name: blending_noise_avo_dncnn_base_T03_avo_mod_seed44_T03_pseudo_deblended_seismic_common_receiver_mod
+  output_dir: /data/data/results/blending_noise_avo/dncnn
+  seed: 44
+  device: cuda
+data:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  loader:
+    batch_size: 142
+    num_workers: 4
+    pin_memory: true
+preprocess:
+  dt: 0.004
+  normalize_mode: max_abs
+  normalize_scope: shot
+  patch_time: 256
+  patch_trace: 128
+  patch_overlap: 0.5
+  max_gathers: null
+  skip: []
+model:
+  type: dncnn
+  params:
+    in_channels: 1
+    out_channels: 1
+    depth: 17
+    base_channels: 64
+    kernel_size: 3
+loss:
+  type: mse
+  params:
+    reduction: mean
+metrics:
+- name: snr
+  params:
+    reduction: per_sample
+- name: psnr
+  params:
+    data_range: 1.0
+    reduction: per_sample
+- name: ssim
+  params:
+    data_range: 2.0
+    window_size: 11
+    sigma: 1.5
+- name: mae
+  params: {}
+- name: mse
+  params: {}
+- name: rmse
+  params:
+    reduction: per_sample
+metrics_filter:
+  snr_min_signal_energy: 1.0e-08
+optim:
+  type: adamw
+  params:
+    lr: 0.0001
+    weight_decay: 1.0e-05
+scheduler:
+  type: cosine
+  params:
+    min_lr: 1.0e-06
+train:
+  epochs: 200
+  grad_clip: 1.0
+  log_step: false
+  log_interval: 10
+  eval_interval: 1
+  ckpt_interval: 20
+  vis_interval: 5
+  resume: null
+log:
+  log_dir: logs
+  plot_interval: 5
+inference:
+  segy_pair:
+    input_path: /data/data/T03_pseudo_deblended_seismic_common_receiver_mod.sgy
+    target_path: /data/data/T03_seismic_common_receiver.sgy
+    traces_per_gather: 1001
+    time_downsample: 1
+  gather_split:
+    train: 96
+    val: 12
+    test: 12
+  checkpoint: /data/data/results/blending_noise_avo/dncnn/blending_noise_avo_dncnn_base/checkpoints/best.pt
+  output_dir: /data/data/results/blending_noise_avo/dncnn/blending_noise_avo_dncnn_base/inference_integrated_metrics
+  n_viz_gathers: 5
+  device: cuda:0
+  batch_size: 142
+  save_npy: false
+  binned_metrics:
+    enabled: true
+    eb_wse:
+      enabled: true
+      bins:
+      - - 5
+        - 20
+      - - 20
+        - 40
+      - - 40
+        - 70
+      - - 70
+        - 100
+      smooth_sigma: 1.0
+    fb_fre:
+      enabled: true
+      rel_threshold: 0.001
+      band_ratios:
+      - 0.2
+      - 0.3
+      - 0.3
+      - 0.2
+      band_names:
+      - low
+      - mid
+      - high
+      - very_high
+      taper_width: null