我的算法描述可以改进吗？

Question

我正在编写凸优化中 FISTA（带线搜索）算法的伪代码描述。（我使用的是幻灯片 17-18这里作为来源。）有人能建议对我的乳胶代码进行任何改进吗？我对任何能改善风格或使演示更清晰、更标准或更美观的东西都感兴趣。这是一个 MWE，输出如下所示：

\documentclass{article}
\usepackage{amsmath,amssymb,algpseudocode,algorithm}
\newcommand{\prox}{\text{prox}}

\begin{document}

\begin{algorithm}[H]
\caption{FISTA with line search}
\label{fistaAlg_lineSearch}
\begin{algorithmic}
\State Initialize $x^0 = v^0$ and $t^0 > 0$
\For{$k = 1,2,\ldots$}
\State $t = t^{k-1}/r$
\State $\text{accept} = \text{FALSE}$
\While{$\text{accept} \neq \text{TRUE}$}
\State $\theta = \begin{cases} 1 & \quad \text{if } k = 1 \\
\text{positive root of } t_{k-1} \theta^2 = t \theta_{k-1}^2(1 - \theta) & \quad \text{if } k > 1 \end{cases}$
\State $y = (1 - \theta) x^{k-1} + \theta v^{k-1}$
\State $x = \prox_{t g}(y - t \nabla f(y))$
\If{$f(x) > f(y) + \langle \nabla f(y),x - y \rangle + \frac{1}{2t} \| x - y \|_2^2$}
\State $t \gets r t$
\Else
\State $\text{accept} = \text{TRUE}$
\EndIf
\EndWhile
\State $t^k = t$
\State $x^k = x$
\State $v^k = x^{k-1} + \frac{1}{\theta}(x^k - x^{k-1})$
\EndFor
\end{algorithmic}
\end{algorithm}

\end{document}