Calculate the mean in a single operation with matrices

Question

Marco Salvo Cottone 2018-4-28

1
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/397869-calculate-the-mean-in-a-single-operation-with-matrices

编辑： Marco Salvo Cottone 2018-4-28

Given 2 matrices M and P: I want to calculate the weighted geometric mean. For a single column vector p i would do like these:

M = rand(3,4);
P = (perms(1:3))';
p_single = P(:,1);                             % And iterate with a for loop through P matrix columns
result = prod(M.^p_single).^(1/sum(p_single)); % This gives me a single line

I want to get Result matrix where each row represents a Mean vector with the corresponding column from P matrix.

How can I do this without iteration in a single or few lines of code ?

EDIT (rewrote the complete loop):

M = rand(3,4);
P = (perms(1:3))';
[~,sz_C] = size(P);
result = [];
for k = 1: sz_C
  p_single = P(:,k);
  mean_single = prod(M.^p_single).^(1/sum(p_single));
  result = [result; mean_single];
end

This code appends to the 'result' matrix every calculated geometric mean combination for P columns.

Is it possible to make it one liner or to eliminate the loop for speed (in case of big data set) ?

4 个评论
显示 2更早的评论隐藏 2更早的评论

dpb 2018-4-28

编辑：dpb 2018-4-28

在 MATLAB Online 中打开

Preallocate and populate instead of augment...since the sizes of M and P aren't commensurate in that size(result) is

size(P,2) x size(M,2)

where P,M aren't directly commensurate since size(P,2)-->factorial(M,1). This means if M gets very big you're going to need a whole bunch of memory real soon now...just the output of perms for N is

>> factorial(10)*8/1024/1024
ans =
            27.685546875
>> factorial(11)*8/1024/1024
ans =
           304.541015625
>>

Marco Salvo Cottone 2018-4-28

编辑：Marco Salvo Cottone 2018-4-28

在 MATLAB Online 中打开

M is static data it will stay small or at least not that big, what will really grow here (when I refer to big data set) is W matrix here named P and the Result matrix will be as you wrote size(P,2) x size(M,2).

I agree I can preallocate:

result = zeros(size(P,2), size(M,2));

But still the for loop problem remains.

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

dpb 2018-4-28

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/397869-calculate-the-mean-in-a-single-operation-with-matrices#answer_317681

在 MATLAB Online 中打开

mngeo=geomean(M.*W);

where M, W are the matrix and associated weights matrix

7 个评论
显示 5更早的评论隐藏 5更早的评论

Marco Salvo Cottone 2018-4-28

编辑：Marco Salvo Cottone 2018-4-28

Each row of M represents a Color expressed as reflectance value at a distance of 10nm each in the visible spectrum of light. I wrote a simple example to make it more easy to follow.

Real M matrix is like 6x36 and Weights/Parts matrix alot bigger, to simulate my requirements think about a W matrix like permutations with repetition with 10^6 columns. (In my example I used different permutation method, but it still applies to the theory, not the function).

To find a subractive color mixing recipe for a color I must create the Weighted Geometric Mean of the 6 rows of M (6 colors at a time) for 10^6 times with the columns of P (each column represent the proportion of parts of those 6 colors), the resulting matrix will give me the mixes, those mixes will be compared to a reference color (the recipe color) after being converting to Lab color space.

You can read it more at Subtractive Color Mixture Computation.

Currently the calculation is too slow because of the loop.

dpb 2018-4-28

"Currently the calculation is too slow because of the loop."

I don't think it's the loop itself that is the bottleneck nearly as much as that you're augmenting the output array every pass instead of preallocating and filling. If it really is calculating what you want and you're absolutely sure of that, fix the above and then see if it isn't adequate--loops are much more well-optimized now in ML than in years gone by but the reallocation within the loop is a real killer.

I'll have to come back and read up a little more on the actual computation; sorry I don't have the time at the moment to do more than just make a quick comment.

Marco Salvo Cottone 2018-4-28