Hugging Face's logo

GeoBrain
/
ground-roll-attenuation

PyTorch
seismic
ground-roll
denoising
unet
resunet
dncnn
attention-unet
Model card Files
xet
 Community
ZDDWLIG commited on
Commit
6459c9c
·
verified ·
1 Parent(s): db249d6

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +25 -5
README.md CHANGED
@@ -38,11 +38,11 @@ Five ground-roll intensity levels produce paired noisy / noise-label records:
38
 
39
  | Level | SNR (dB) | PSNR (dB) | SSIM | MAE | MSE | RMSE |
40
  |-------|----------|-----------|------|-----|-----|------|
41
- | 1.0 | -2.7129 | 19.1193 | 0.6766 | 0.049940 | 0.012248 | 0.110671 |
42
- | 3.0 | 6.8295 | 25.1399 | 0.8047 | 0.024970 | 0.003062 | 0.055335 |
43
- | 5.0 | 11.2665 | 28.6618 | 0.8604 | 0.016647 | 0.001361 | 0.036890 |
44
- | 7.0 | 14.1891 | 31.1605 | 0.8929 | 0.012485 | 0.000766 | 0.027668 |
45
- | 9.0 | 16.3720 | 33.0987 | 0.9149 | 0.009988 | 0.000490 | 0.022134 |
46
 
47
  *Metrics computed on the test set (2D flattened shot gathers) in the normalized domain before denoising.*
48
 
@@ -122,6 +122,10 @@ Mean ± std over 3 seeds on the held-out test shot (FFID=9), evaluated on 2D-fla
122
  | Method | Params (M) | SNR (dB) | PSNR (dB) | SSIM | MAE | MSE | RMSE |
123
  | --- | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
124
  | Raw (noisy) | — | 2.7129±0.0000 | 21.8322±0.0000 | 0.9527±0.0000 | 0.015312±0.000000 | 0.006558±0.000000 | 0.080982±0.000000 |
 
 
 
 
125
  | Enhanced Atten-UNet | 8.2 | 11.9244±0.4043 | 33.7566±0.4043 | 0.9288±0.0024 | 0.008774±0.000263 | 0.000422±0.000039 | 0.020535±0.000949 |
126
  | SANet | 0.23 | 21.4884±0.3003 | 40.6078±0.3002 | 0.9886±0.0009 | 0.003088±0.000175 | 0.000087±0.000006 | 0.009328±0.000324 |
127
  | Physics CNN | 7.27 | 22.3319±0.5078 | 36.2584±0.5077 | 0.9728±0.0028 | 0.006262±0.000573 | 0.000238±0.000027 | 0.015402±0.000889 |
@@ -133,6 +137,10 @@ Mean ± std over 3 seeds on the held-out test shot (FFID=9), evaluated on 2D-fla
133
  | Method | Params (M) | SNR (dB) | PSNR (dB) | SSIM | MAE | MSE | RMSE |
134
  | --- | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
135
  | Raw (noisy) | — | -6.8295±0.0000 | 18.3104±0.0000 | 0.9477±0.0000 | 0.022968±0.000000 | 0.014756±0.000000 | 0.121473±0.000000 |
 
 
 
 
136
  | Enhanced Atten-UNet | 8.2 | 18.5837±0.9329 | 36.8940±0.9330 | 0.9580±0.0027 | 0.006279±0.000549 | 0.000208±0.000045 | 0.014354±0.001559 |
137
  | SANet | 0.23 | 17.1129±0.2490 | 42.2528±0.2490 | 0.9876±0.0006 | 0.002294±0.000126 | 0.000060±0.000003 | 0.007717±0.000222 |
138
  | Physics CNN | 7.27 | 21.2933±0.2677 | 36.0215±0.2677 | 0.9613±0.0015 | 0.007664±0.000260 | 0.000250±0.000016 | 0.015815±0.000491 |
@@ -144,6 +152,10 @@ Mean ± std over 3 seeds on the held-out test shot (FFID=9), evaluated on 2D-fla
144
  | Method | Params (M) | SNR (dB) | PSNR (dB) | SSIM | MAE | MSE | RMSE |
145
  | --- | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
146
  | Raw (noisy) | — | -11.2665±0.0000 | 17.3952±0.0000 | 0.9466±0.0000 | 0.025520±0.000000 | 0.018217±0.000000 | 0.134970±0.000000 |
 
 
 
 
147
  | Enhanced Atten-UNet | 8.2 | 21.4265±0.8323 | 38.8217±0.8323 | 0.9708±0.0019 | 0.004890±0.000354 | 0.000133±0.000024 | 0.011488±0.001072 |
148
  | SANet | 0.23 | 14.2619±0.1390 | 42.9236±0.1390 | 0.9875±0.0004 | 0.001949±0.000145 | 0.000051±0.000002 | 0.007142±0.000115 |
149
  | Physics CNN | 7.27 | 21.8948±0.1265 | 36.8837±0.1264 | 0.9690±0.0001 | 0.006944±0.000087 | 0.000205±0.000006 | 0.014317��0.000209 |
@@ -155,6 +167,10 @@ Mean ± std over 3 seeds on the held-out test shot (FFID=9), evaluated on 2D-fla
155
  | Method | Params (M) | SNR (dB) | PSNR (dB) | SSIM | MAE | MSE | RMSE |
156
  | --- | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
157
  | Raw (noisy) | — | -14.1891±0.0000 | 16.9715±0.0000 | 0.9461±0.0000 | 0.026796±0.000000 | 0.020084±0.000000 | 0.141719±0.000000 |
 
 
 
 
158
  | Enhanced Atten-UNet | 8.2 | 23.4293±1.1926 | 40.4008±1.1926 | 0.9785±0.0012 | 0.003988±0.000376 | 0.000093±0.000023 | 0.009608±0.001265 |
159
  | SANet | 0.23 | 12.0953±0.1912 | 43.2559±0.1912 | 0.9877±0.0008 | 0.001822±0.000184 | 0.000047±0.000002 | 0.006875±0.000151 |
160
  | Physics CNN | 7.27 | 22.2912±0.4163 | 37.4043±0.4163 | 0.9701±0.0034 | 0.006623±0.000452 | 0.000182±0.000018 | 0.013494±0.000653 |
@@ -166,6 +182,10 @@ Mean ± std over 3 seeds on the held-out test shot (FFID=9), evaluated on 2D-fla
166
  | Method | Params (M) | SNR (dB) | PSNR (dB) | SSIM | MAE | MSE | RMSE |
167
  | --- | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
168
  | Raw (noisy) | — | -16.3720±0.0000 | 16.7268±0.0000 | 0.9458±0.0000 | 0.027561±0.000000 | 0.021248±0.000000 | 0.145768±0.000000 |
 
 
 
 
169
  | Enhanced Atten-UNet | 8.2 | 24.8580±1.5442 | 41.5847±1.5442 | 0.9820±0.0028 | 0.003487±0.000463 | 0.000072±0.000023 | 0.008417±0.001415 |
170
  | SANet | 0.23 | 10.5793±0.1366 | 43.6780±0.1367 | 0.9884±0.0006 | 0.001715±0.000084 | 0.000043±0.000001 | 0.006548±0.000104 |
171
  | Physics CNN | 7.27 | 23.0216±0.3917 | 38.2117±0.3917 | 0.9726±0.0024 | 0.006038±0.000219 | 0.000151±0.000013 | 0.012295±0.000550 |
 
38
 
39
  | Level | SNR (dB) | PSNR (dB) | SSIM | MAE | MSE | RMSE |
40
  |-------|----------|-----------|------|-----|-----|------|
41
+ | 1.0 | 2.7129 | 21.8322 | 0.9527 | 0.015312 | 0.006558 | 0.080982 |
42
+ | 3.0 | -6.8295 | 18.3104 | 0.9477 | 0.022968 | 0.014756 | 0.121473 |
43
+ | 5.0 | -11.2665 | 17.3952 | 0.9466 | 0.025520 | 0.018217 | 0.134970 |
44
+ | 7.0 | -14.1891 | 16.9715 | 0.9461 | 0.026796 | 0.020084 | 0.141719 |
45
+ | 9.0 | -16.3720 | 16.7268 | 0.9458 | 0.027561 | 0.021248 | 0.145768 |
46
 
47
  *Metrics computed on the test set (2D flattened shot gathers) in the normalized domain before denoising.*
48
 
 
122
  | Method | Params (M) | SNR (dB) | PSNR (dB) | SSIM | MAE | MSE | RMSE |
123
  | --- | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
124
  | Raw (noisy) | — | 2.7129±0.0000 | 21.8322±0.0000 | 0.9527±0.0000 | 0.015312±0.000000 | 0.006558±0.000000 | 0.080982±0.000000 |
125
+ | UNet | 7.76 | 28.3886±1.0929 | 47.5080±1.0930 | 0.9972±0.0004 | 0.000673±0.000129 | 0.000018±0.000004 | 0.004235±0.000524 |
126
+ | ResUNet | 8.11 | 31.6226±0.5664 | 50.7420±0.5664 | 0.9980±0.0001 | 0.000659±0.000105 | 0.000008±0.000001 | 0.002907±0.000186 |
127
+ | DnCNN | 0.14 | 19.7355±0.3250 | 38.8548±0.3250 | 0.9846±0.0014 | 0.004684±0.000236 | 0.000130±0.000010 | 0.011415±0.000425 |
128
+ | Attention UNet | 7.85 | 29.4786±1.3513 | 48.5979±1.3514 | 0.9974±0.0007 | 0.000577±0.000109 | 0.000014±0.000004 | 0.003745±0.000557 |
129
  | Enhanced Atten-UNet | 8.2 | 11.9244±0.4043 | 33.7566±0.4043 | 0.9288±0.0024 | 0.008774±0.000263 | 0.000422±0.000039 | 0.020535±0.000949 |
130
  | SANet | 0.23 | 21.4884±0.3003 | 40.6078±0.3002 | 0.9886±0.0009 | 0.003088±0.000175 | 0.000087±0.000006 | 0.009328±0.000324 |
131
  | Physics CNN | 7.27 | 22.3319±0.5078 | 36.2584±0.5077 | 0.9728±0.0028 | 0.006262±0.000573 | 0.000238±0.000027 | 0.015402±0.000889 |
 
137
  | Method | Params (M) | SNR (dB) | PSNR (dB) | SSIM | MAE | MSE | RMSE |
138
  | --- | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
139
  | Raw (noisy) | — | -6.8295±0.0000 | 18.3104±0.0000 | 0.9477±0.0000 | 0.022968±0.000000 | 0.014756±0.000000 | 0.121473±0.000000 |
140
+ | UNet | 7.76 | 23.0729±0.0993 | 48.2128±0.0994 | 0.9961±0.0001 | 0.000676±0.000022 | 0.000015±0.000000 | 0.003885±0.000044 |
141
+ | ResUNet | 8.11 | 22.6469±0.9303 | 47.7869±0.9303 | 0.9924±0.0015 | 0.001194±0.000660 | 0.000017±0.000003 | 0.004095±0.000424 |
142
+ | DnCNN | 0.14 | 17.2389±0.2322 | 42.3788±0.2322 | 0.9883±0.0008 | 0.002676±0.000161 | 0.000058±0.000003 | 0.007606±0.000205 |
143
+ | Attention UNet | 7.85 | 24.3939±1.2177 | 49.5338±1.2177 | 0.9964±0.0011 | 0.000630±0.000141 | 0.000011±0.000003 | 0.003358±0.000453 |
144
  | Enhanced Atten-UNet | 8.2 | 18.5837±0.9329 | 36.8940±0.9330 | 0.9580±0.0027 | 0.006279±0.000549 | 0.000208±0.000045 | 0.014354±0.001559 |
145
  | SANet | 0.23 | 17.1129±0.2490 | 42.2528±0.2490 | 0.9876±0.0006 | 0.002294±0.000126 | 0.000060±0.000003 | 0.007717±0.000222 |
146
  | Physics CNN | 7.27 | 21.2933±0.2677 | 36.0215±0.2677 | 0.9613±0.0015 | 0.007664±0.000260 | 0.000250±0.000016 | 0.015815±0.000491 |
 
152
  | Method | Params (M) | SNR (dB) | PSNR (dB) | SSIM | MAE | MSE | RMSE |
153
  | --- | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
154
  | Raw (noisy) | — | -11.2665±0.0000 | 17.3952±0.0000 | 0.9466±0.0000 | 0.025520±0.000000 | 0.018217±0.000000 | 0.134970±0.000000 |
155
+ | UNet | 7.76 | 20.2195±0.3210 | 48.8813±0.3210 | 0.9958±0.0001 | 0.000656±0.000033 | 0.000013±0.000001 | 0.003599±0.000133 |
156
+ | ResUNet | 8.11 | 18.1078±2.1170 | 46.7695±2.1170 | 0.9898±0.0021 | 0.001265±0.000143 | 0.000023±0.000011 | 0.004679±0.001134 |
157
+ | DnCNN | 0.14 | 15.5662±0.2235 | 44.2280±0.2236 | 0.9898±0.0006 | 0.002042±0.000221 | 0.000038±0.000002 | 0.006147±0.000159 |
158
+ | Attention UNet | 7.85 | 21.9546±1.0678 | 50.6163±1.0678 | 0.9965±0.0011 | 0.000618±0.000068 | 0.000009±0.000002 | 0.002961±0.000359 |
159
  | Enhanced Atten-UNet | 8.2 | 21.4265±0.8323 | 38.8217±0.8323 | 0.9708±0.0019 | 0.004890±0.000354 | 0.000133±0.000024 | 0.011488±0.001072 |
160
  | SANet | 0.23 | 14.2619±0.1390 | 42.9236±0.1390 | 0.9875±0.0004 | 0.001949±0.000145 | 0.000051±0.000002 | 0.007142±0.000115 |
161
  | Physics CNN | 7.27 | 21.8948±0.1265 | 36.8837±0.1264 | 0.9690±0.0001 | 0.006944±0.000087 | 0.000205±0.000006 | 0.014317��0.000209 |
 
167
  | Method | Params (M) | SNR (dB) | PSNR (dB) | SSIM | MAE | MSE | RMSE |
168
  | --- | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
169
  | Raw (noisy) | — | -14.1891±0.0000 | 16.9715±0.0000 | 0.9461±0.0000 | 0.026796±0.000000 | 0.020084±0.000000 | 0.141719±0.000000 |
170
+ | UNet | 7.76 | 17.8984±0.2809 | 49.0590±0.2809 | 0.9955±0.0003 | 0.000620±0.000054 | 0.000012±0.000001 | 0.003526±0.000114 |
171
+ | ResUNet | 8.11 | 16.6023±1.4399 | 47.7628±1.4399 | 0.9895±0.0039 | 0.001264±0.000417 | 0.000017±0.000005 | 0.004128±0.000649 |
172
+ | DnCNN | 0.14 | 15.1741±0.1268 | 46.3346±0.1268 | 0.9925±0.0003 | 0.001628±0.000134 | 0.000023±0.000001 | 0.004823±0.000071 |
173
+ | Attention UNet | 7.85 | 17.9454±2.7925 | 49.1059±2.7925 | 0.9947±0.0031 | 0.000686±0.000241 | 0.000014±0.000008 | 0.003623±0.001098 |
174
  | Enhanced Atten-UNet | 8.2 | 23.4293±1.1926 | 40.4008±1.1926 | 0.9785±0.0012 | 0.003988±0.000376 | 0.000093±0.000023 | 0.009608±0.001265 |
175
  | SANet | 0.23 | 12.0953±0.1912 | 43.2559±0.1912 | 0.9877±0.0008 | 0.001822±0.000184 | 0.000047±0.000002 | 0.006875±0.000151 |
176
  | Physics CNN | 7.27 | 22.2912±0.4163 | 37.4043±0.4163 | 0.9701±0.0034 | 0.006623±0.000452 | 0.000182±0.000018 | 0.013494±0.000653 |
 
182
  | Method | Params (M) | SNR (dB) | PSNR (dB) | SSIM | MAE | MSE | RMSE |
183
  | --- | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
184
  | Raw (noisy) | — | -16.3720±0.0000 | 16.7268±0.0000 | 0.9458±0.0000 | 0.027561±0.000000 | 0.021248±0.000000 | 0.145768±0.000000 |
185
+ | UNet | 7.76 | 16.5997±0.4636 | 49.6984±0.4636 | 0.9956±0.0005 | 0.000673±0.000096 | 0.000011±0.000001 | 0.003277±0.000177 |
186
+ | ResUNet | 8.11 | 13.8944±1.2448 | 46.9932±1.2448 | 0.9873±0.0053 | 0.001799±0.000734 | 0.000021±0.000006 | 0.004502±0.000670 |
187
+ | DnCNN | 0.14 | 14.8737±0.1844 | 47.9725±0.1845 | 0.9942±0.0003 | 0.001366±0.000120 | 0.000016±0.000001 | 0.003994±0.000085 |
188
+ | Attention UNet | 7.85 | 15.7861±2.7603 | 48.8849±2.7604 | 0.9940±0.0033 | 0.000822±0.000303 | 0.000015±0.000008 | 0.003715±0.001128 |
189
  | Enhanced Atten-UNet | 8.2 | 24.8580±1.5442 | 41.5847±1.5442 | 0.9820±0.0028 | 0.003487±0.000463 | 0.000072±0.000023 | 0.008417±0.001415 |
190
  | SANet | 0.23 | 10.5793±0.1366 | 43.6780±0.1367 | 0.9884±0.0006 | 0.001715±0.000084 | 0.000043±0.000001 | 0.006548±0.000104 |
191
  | Physics CNN | 7.27 | 23.0216±0.3917 | 38.2117±0.3917 | 0.9726±0.0024 | 0.006038±0.000219 | 0.000151±0.000013 | 0.012295±0.000550 |