注水法

注水算法的基本原理就是根据香农公式和限制条件，通过拉格朗日乘数法组成的一个方程，先令其偏导为零，求出Pi的表达式，但是Pi的表达式中包含一个未知数，再根据限制条件可以先求解出该未知数，再回代到之前的方程中，可以求解得每个信道根据信道质量分配得到的Pi。求解问题如下：
mimo信道容量： max: $C=\sum _{i=1}^m{\log }_2\left(1+\frac{P_i}{{\sigma }^2}{\lambda }_i\right)$
功率满足： $\sum _{i=1}^m{P}_i=P$
上面为等式约束问题，可用拉格朗日乘子法求解。
引入拉格朗日函数，有 $Z(\lambda ,P_i)=\sum _{i=1}^m{\log }_2\left(1+\frac{P_i}{{\sigma }^2}{\lambda }_i\right)+L(P-\sum _{i=1}^m{P}_i)$
对$P_i$求偏导，有

解得：$P_i=\frac{1}{L\bullet \ln 2}-\frac{{\sigma }^2}{{\lambda }_i}=\mu -\frac{{\sigma }^2}{{\lambda }_i}$
其中$\mu$为常数
因为功率不能为负数，将上述写为如下形式：
$P_i={\left(\mu -\frac{{\sigma }^2}{{\lambda }_i}\right)}^{+}$
所以最优的分配策略如上式
其中${\left(a\right)}^{+}=\max (a,0)$
这里可用通过$$\sum _{i=1}^m{P}_i=P=m\mu -\sum _{i=1}^m\frac{{\sigma }^2}{{\lambda }^2}$$
得到：$\mu =\frac{P+\sum _{i=1}^m\frac{{\sigma }^2}{{\lambda }^2}}{m}$

也就是说，当已知csi（信道状态信息时），发送端会选择好的信道，给它分配大功率，过程如图示