ํผ๋๋ก ๋์๊ฐ๊ธฐ
Dev.toAI/ML
์๋ฌธ ์ฝ๊ธฐ
Bayesian NN๊ณผ Modulo-256 ๊ธฐ๋ฐ 100% ๊ฐ์ญ์ ์คํ ๊ฐ๋ ธ๊ทธ๋ํผ ๊ตฌํ
Perfect Pixel Recovery: Implementing Bayesian Neural Networks for Reversible Steganography ๐ต๏ธโโ๏ธ
AI ์์ฝ
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 ํจ์ ์ ์ฉ ๊ณ ๋ ค
ํ๊ทธ