ํ”ผ๋“œ๋กœ ๋Œ์•„๊ฐ€๊ธฐ
Perfect Pixel Recovery: Implementing Bayesian Neural Networks for Reversible Steganography ๐Ÿ•ต๏ธโ€โ™‚๏ธ
Dev.toDev.to
AI/ML

Bayesian NN๊ณผ Modulo-256 ๊ธฐ๋ฐ˜ 100% ๊ฐ€์—ญ์  ์Šคํ…Œ๊ฐ€๋…ธ๊ทธ๋ž˜ํ”ผ ๊ตฌํ˜„

Perfect Pixel Recovery: Implementing Bayesian Neural Networks for Reversible Steganography ๐Ÿ•ต๏ธโ€โ™‚๏ธ

Anjasfedo2026๋…„ 5์›” 19์ผ4๋ถ„advanced

Context

๊ธฐ์กด Steganography ๋ฐฉ์‹์˜ ํ”ฝ์…€ Overflow ๋ฐ Underflow๋กœ ์ธํ•œ Clipping ํ˜„์ƒ์œผ๋กœ ๋ฐ์ดํ„ฐ ์†์‹ค ๋ฐœ์ƒ. ์ˆ˜์‹  ์ธก์—์„œ ์›๋ณธ ์ด๋ฏธ์ง€๋ฅผ ํ”ฝ์…€ ๋‹จ์œ„๋กœ ์™„์ „ ๋ณต๊ตฌํ•˜๋Š” Reversible Steganography ๊ตฌํ˜„์˜ ๊ธฐ์ˆ ์  ์ œ์•ฝ ์กด์žฌ.

Technical Solution

  • Uncertainty Estimation์„ ํ†ตํ•œ Variance Map ์ƒ์„ฑ์œผ๋กœ ์‹œ๊ฐ์  ์™œ๊ณก ์ตœ์†Œํ™” ์˜์—ญ์œผ๋กœ ๋ฐ์ดํ„ฐ ์ž„๋ฒ ๋”ฉ ์ตœ์ ํ™”.
  • Bayesian Neural Network๋ฅผ ๋„์ž…ํ•˜์—ฌ ํ”ฝ์…€ ๊ฐ•๋„ ์˜ˆ์ธก๊ฐ’๊ณผ ์˜ˆ์ธก ๋ถˆํ™•์‹ค์„ฑ(Variance)์„ ๋™์‹œ ์‚ฐ์ถœ.
  • Modulo-256 Arithmetic ์ ์šฉ์„ ํ†ตํ•œ ํ”ฝ์…€ ๊ฐ’ ๋ฒ”์œ„ ์ดˆ๊ณผ ์‹œ Wrapping ์ฒ˜๋ฆฌ๋กœ Clipping ๋ฌธ์ œ ์›์ฒœ ํ•ด๊ฒฐ.
  • Uncertainty-weighted Loss ํ•จ์ˆ˜ ์„ค๊ณ„๋ฅผ ํ†ตํ•ด ๋ชจ๋ธ์˜ ์‹ ๋ขฐ๋„ ๊ธฐ๋ฐ˜ ๊ฐ€์ค‘์น˜ ์ ์šฉ ๋ฐ ํ•™์Šต ์•ˆ์ •์„ฑ ํ™•๋ณด.
  • Checkerboard Pattern ๊ธฐ๋ฐ˜์˜ ๋™์ผ NN Context ๊ณต์œ ๋ฅผ ํ†ตํ•œ ์†ก์ˆ˜์‹  ๊ฐ„ ์ •๋ฐ€ํ•œ Modulo Difference ๊ณ„์‚ฐ ๋ฐ ๋ณต์›.

1. ๋ฐ์ดํ„ฐ ์†์‹ค์ด ๋ถˆ๊ฐ€ํ•œ ๊ฐ€์—ญ์  ์‹œ์Šคํ…œ ์„ค๊ณ„ ์‹œ Clipping ๋ฐฉ์ง€๋ฅผ ์œ„ํ•œ Modulo ์—ฐ์‚ฐ ๋„์ž… ๊ฒ€ํ† 

2. ๋‹จ์ˆœ ์˜ˆ์ธก๊ฐ’์ด ์•„๋‹Œ ์˜ˆ์ธก์˜ ์‹ ๋ขฐ๋„(Variance)๋ฅผ ํ™œ์šฉํ•œ ๋ฐ์ดํ„ฐ ๋ฐฐ์น˜ ์ „๋žต ์ˆ˜๋ฆฝ

3. ์†์‹ค ํ•จ์ˆ˜ ์„ค๊ณ„ ์‹œ MSE ๋Œ€์‹  ๋„๋ฉ”์ธ ํŠน์„ฑ์— ๋งž๋Š” ๊ฐ€์ค‘์น˜ ๊ธฐ๋ฐ˜ Loss ํ•จ์ˆ˜ ์ ์šฉ ๊ณ ๋ ค

์›๋ฌธ ์ฝ๊ธฐ