\begin{tikzpicture}

\tikzstyle{h2o} = [or, mark=diamond, mark size=1pt]
\tikzstyle{snap} = [gr, mark=x, mark size=1.3pt]
\tikzstyle{pyramid} = [bl, mark=+, mark size=1.3pt]
\tikzstyle{prob} = [red, mark=*, mark size=0.7pt]

\begin{groupplot}[group style={columns=3, horizontal sep=1.2cm, vertical sep=1.2cm},
    width=5.0cm,
    height=4.0cm,
    every axis plot/.append style={thick},
    % Label
    xlabel shift=-0.08cm,         
    ylabel shift=-0.15cm,
    xlabel near ticks,
    ylabel near ticks,
    label style={font=\scriptsize},
    xlabel={KV cache ratio},
    ylabel={Rel. performance},
    % Grid
    grid=major,
    xmajorgrids=true,
    ymajorgrids=true,
    major grid style={dotted, black},
    % tick
    tick label style={font=\scriptsize},
    tick pos=left,
    x tick label style={/pgf/number format/.cd, fixed, fixed zerofill, precision=1},
    y tick label style={/pgf/number format/.cd, fixed, fixed zerofill, precision=0},
    xmax=1.0,
    xmin=0.1,
    xtick={0.2, 0.4, 0.6, 0.8, 1.0},
    extra x ticks={0.3, 0.5, 0.7, 0.9},   
    extra x tick labels={,,,},
    extra x tick style={
        grid=none,
        tick style={thin},
        major tick length=2.4pt,
    },    
    % title
    title style={
      at={(axis description cs:0.5,0.93)}, 
      anchor=south,
      font={\footnotesize}
    },    
]

%%%% Retrieval-Matching %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

\nextgroupplot[title=NIAH, ylabel={Accuracy (\%)}, ytick distance=20]
\addplot[prob] table[x=ratio, y=needle-prob, col sep=comma]{data/llama-3b.csv};
\addplot[h2o] table[x=ratio, y=needle-h2o, col sep=comma]{data/llama-3b.csv};
\addplot[snap] table[x=ratio, y=needle-snap, col sep=comma]{data/llama-3b.csv};
\addplot[pyramid] table[x=ratio, y=needle-pyramid, col sep=comma]{data/llama-3b.csv};

\nextgroupplot[title=SQuAD, ylabel={Accuracy (\%)}, ymax=100,
legend columns=4, legend style={at={(0.5,1.25)}, anchor=south, font=\footnotesize},
]
\addplot[prob] table[x=ratio, y=squad-prob, col sep=comma]{data/llama-3b.csv};\addlegendentry{KVzip (ours)}
\addplot[h2o] table[x=ratio, y=squad-h2o, col sep=comma]{data/llama-3b.csv};\addlegendentry{$\text{H}_2\text{O}$}
\addplot[snap] table[x=ratio, y=squad-snap, col sep=comma]{data/llama-3b.csv};\addlegendentry{SnapKV}
\addplot[pyramid] table[x=ratio, y=squad-pyramid, col sep=comma]{data/llama-3b.csv};\addlegendentry{PyramidKV}

\nextgroupplot[title=GSM8K, ylabel={Accuracy (\%)}]
\addplot[h2o] table[x=ratio, y=gsm-h2o, col sep=comma]{data/llama-3b.csv};
\addplot[snap] table[x=ratio, y=gsm-snap, col sep=comma]{data/llama-3b.csv};
\addplot[pyramid] table[x=ratio, y=gsm-pyramid, col sep=comma]{data/llama-3b.csv};
\addplot[prob] table[x=ratio, y=gsm-prob, col sep=comma]{data/llama-3b.csv};


\end{groupplot}
\end{tikzpicture}
