diff --git a/fig/fig0_overview2.pdf b/fig/fig0_overview2.pdf index aa31fba..25ac334 100755 Binary files a/fig/fig0_overview2.pdf and b/fig/fig0_overview2.pdf differ diff --git a/fig/fig0_overview2_tikz.tex b/fig/fig0_overview2_tikz.tex index 27df9eb..21e34cf 100755 --- a/fig/fig0_overview2_tikz.tex +++ b/fig/fig0_overview2_tikz.tex @@ -5,11 +5,11 @@ \begin{tikzpicture}[ font=\small, box/.style={draw,rounded corners=1pt,minimum width=7.5mm,minimum height=5.5mm,inner sep=1.5pt}, - enc/.style={box,fill=yellow!35}, + enc/.style={box}, chan/.style={draw,diamond,aspect=1.4,fill=gray!25,inner sep=0.8pt}, rxb/.style={box}, mult/.style={draw,circle,inner sep=0.4pt,minimum size=4mm}, - sum/.style={draw,rounded corners=2.5mm,fill=cyan!20,minimum width=12mm,minimum height=36mm}, + sum/.style={draw,rounded corners=2.5mm,fill=gray!15,minimum width=12mm,minimum height=36mm}, dec/.style={draw,rounded corners=2pt,minimum width=53mm,minimum height=6mm,inner sep=2pt}, lbl/.style={font=\small\bfseries,align=center}, arr/.style={-stealth,semithick}] @@ -54,16 +54,16 @@ \node[lbl] at ([yshift=-4.5mm]sig.south) {Received Signal\\(Superposition $\tilde{\mathbf{y}}$)}; % ---------------- right panel: UWCA decoder ---------------- -\node[dec,fill=violet!15,right=26mm of sig.east,yshift=19mm] (mask) +\node[dec,fill=gray!15,right=26mm of sig.east,yshift=19mm] (mask) {Soft Masks $\{\tilde{\mathbf{m}}_i\}_{i=1}^{U}$}; -\node[dec,fill=blue!12,below=2.2mm of mask,align=center] (proj) +\node[dec,fill=gray!15,below=2.2mm of mask,align=center] (proj) {K/V Projection $(\mathbf{W}_K,\mathbf{W}_V)$\\[-1pt] Queries $\{\mathbf{q}_u\}$}; -\node[dec,fill=orange!30,below=2.2mm of proj,minimum height=9mm,align=center] (attn) +\node[dec,fill=gray!30,below=2.2mm of proj,minimum height=9mm,align=center] (attn) {Scaled Dot-Product Attention\\[-1pt] {\footnotesize $\alpha_{u,i}=\mathrm{softmax}\big(\eta\,\mathbf{q}_u^{\top}\mathbf{k}_i/\sqrt{d_k}\big)$}}; \node[mult,below=2.2mm of attn] (plus) {$\oplus$}; -\node[dec,fill=green!20,below=2.2mm of plus] (norm) {$\ell_2$-Normalize}; -\node[dec,fill=green!30,below=2.2mm of norm] (out) +\node[dec,fill=gray!15,below=2.2mm of plus] (norm) {$\ell_2$-Normalize}; +\node[dec,fill=gray!15,below=2.2mm of norm] (out) {$\hat{\mathbf{e}}_1,\ \hat{\mathbf{e}}_2,\ \ldots,\ \hat{\mathbf{e}}_U$}; \draw[arr] (mask) -- (proj); \draw[arr] (proj) -- (attn); @@ -71,21 +71,21 @@ \draw[arr] (plus) -- (norm); \draw[arr] (norm) -- (out); \draw[arr] (sig.east) -- node[above,pos=0.45,font=\small]{$\tilde{\mathbf{y}}$} ++(13mm,0) |- (mask.west); -\draw[arr,dashed,blue] (mask.east) -- ++(4mm,0) |- (plus.east) +\draw[arr,dashed] (mask.east) -- ++(4mm,0) |- (plus.east) node[pos=0.78,above,font=\footnotesize]{skip: $\tilde{\mathbf{y}}\odot\tilde{\mathbf{m}}_u$}; % decoder panel frame \begin{scope}[on background layer] -\node[rounded corners=2mm,fill=gray!12,fit=(mask)(out)(attn), +\node[rounded corners=2mm,fill=gray!7,fit=(mask)(out)(attn), inner xsep=7mm,inner ysep=4.5mm] (panel) {}; \end{scope} \node[anchor=south,font=\small\bfseries\color{black!75},inner sep=2pt] at (panel.north) {UWCA Decoder (per-user, shared weights $\theta$)}; % MAML brace -\draw[decorate,decoration={brace,mirror,amplitude=2mm},blue!60!black,thick] +\draw[decorate,decoration={brace,mirror,amplitude=2mm},black!75,thick] ([yshift=-1.5mm]panel.south west) -- ([yshift=-1.5mm]panel.south east) - node[midway,below=2mm,font=\small\color{blue!60!black}] + node[midway,below=2mm,font=\small\color{black!75}] {MAML meta-training over SNR tasks $\{\mathcal{T}_k\}_{k=1}^{K}$}; \end{tikzpicture} \end{document}