k-svd算法matlab代码

以下是一个简单的 MATLAB 实现 K-SVD 算法的代码：

function [D,X] = ksvd(Y,D,X,iterations)
% Y: 数据矩阵, d x n
% D: 初始字典, d x k
% X: 初始稀疏表示, k x n
% iterations: 迭代次数

[d,n] = size(Y);
k = size(D,2);

for iter = 1:iterations
    % 更新稀疏表示
    for i = 1:n
        Xi = X(:,i);
        Di = D;
        Di(:,Xi~=0) = 0;
        residual = Y(:,i) - D*Xi;
        active_idx = find(Xi~=0);
        if ~isempty(active_idx)
            residual = residual + D(:,active_idx)*(Xi(active_idx)*active_idx');
        end
        [Ci,~,~] = svds(residual,1);
        Xi = zeros(k,1);
        Xi(active_idx) = X(active_idx,i);
        Xi(Xi~=0) = Ci;
        X(:,i) = Xi;
    end
    
    % 更新字典
    for j = 1:k
        active_idx = find(X(j,:)~=0);
        if ~isempty(active_idx)
            E = Y(:,active_idx) - D*X(:,active_idx) + D(:,j)*X(j,active_idx);
            [U,S,V] = svds(E,1);
            D(:,j) = U;
            X(j,active_idx) = S*V';
        end
    end
end
end

这里的 `svds` 函数用来计算矩阵的奇异值分解。需要注意的是，这个实现并不是最优化的，仅供参考。